如何使用python将CSV文件中的列转换为json，以便键和值对来自CSV的不同列？

import csv
import json

def csv_to_json(csv_file):
    # 打开CSV文件并读取数据
    with open(csv_file, 'r') as file:
        csv_data = list(csv.reader(file))

    # 创建空的JSON对象
    json_data = {}

    # 获取列名
    headers = csv_data[0]

    # 遍历CSV数据列表
    for row in csv_data[1:]:
        # 创建一个字典，用于存储每一行数据
        row_data = {}

        # 遍历每一列数据
        for i, value in enumerate(row):
            # 将列名作为键，对应的值作为值，添加到字典中
            row_data[headers[i]] = value

        # 将每一行数据添加到JSON对象中
        json_data[row[0]] = row_data

    # 将JSON对象转换为字符串格式
    json_str = json.dumps(json_data)

    return json_str

# 示例用法
csv_file = 'data.csv'
json_str = csv_to_json(csv_file)
print(json_str)