欢迎访问宙启技术站
智能推送

将CSV文件转换为JSON格式 - convert_csv_to_json()

发布时间:2023-11-13 06:50:53

要将CSV文件转换为JSON格式,可以使用Python中的csv模块和json模块。以下是一个名为convert_csv_to_json()的函数,可实现将CSV文件转换为JSON格式。

import csv
import json

def convert_csv_to_json(csv_file_path, json_file_path):
    # 读取CSV文件
    with open(csv_file_path, 'r') as csv_file:
        csv_data = csv.DictReader(csv_file)
        # 将CSV数据转换为字典列表
        json_data = [row for row in csv_data]

    # 写入JSON文件
    with open(json_file_path, 'w') as json_file:
        json_file.write(json.dumps(json_data, indent=4))

# 示例用法
csv_file_path = 'input.csv'
json_file_path = 'output.json'
convert_csv_to_json(csv_file_path, json_file_path)

在这个函数中,首先使用csv.DictReader()函数读取CSV文件。csv.DictReader()可以将CSV文件的每一行数据转换为一个字典,其中字典的键是CSV文件的列头,值是对应行的数据。

然后,使用列表推导式将CSV数据转换为字典列表。

最后,使用json.dumps()函数将字典列表转换为JSON格式,并使用json_file.write()函数将JSON数据写入到JSON文件中。indent=4参数用于在JSON文件中进行缩进,使其更易读。

可以通过传入CSV文件路径和JSON文件路径来调用convert_csv_to_json()函数,并将CSV文件转换为JSON格式。

注意:在使用此函数之前,确保已安装了Python的csv和json模块。

希望这可以帮助到你!