将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模块。
希望这可以帮助到你!
