以下是一个将字典列表写入CSV的函数的代码示例:
import csv
def write_dict_list_to_csv(dict_list, filename):
# 获取所有字典的键,作为CSV文件的表头
headers = list(set().union(*(d.keys() for d in dict_list)))
# 使用csv模块打开文件,并写入字典列表
with open(filename, 'w', newline='') as csvfile:
writer = csv.DictWriter(csvfile, fieldnames=headers)
# 写入表头
writer.writeheader()
# 逐行写入字典数据
for row in dict_list:
writer.writerow(row)
# 示例用法
data = [
{'Name': 'John', 'Age': 25, 'City': 'New York'},
{'Name': 'Emily', 'Age': 30, 'City': 'Los Angeles'},
{'Name': 'Michael', 'Age': 35, 'City': 'Chicago'}
]
write_dict_list_to_csv(data, 'output.csv')
这个函数接受两个参数,dict_list
是一个字典列表,filename
是要写入的CSV文件的名称。函数首先获取所有字典的键作为CSV文件的表头,然后使用csv
模块打开文件并创建一个DictWriter
对象。然后,函数逐行写入字典数据到CSV文件中。
在示例中,我们定义了一个包含三个字典的列表data
,每个字典表示一个人的信息。然后我们调用write_dict_list_to_csv
函数将这个列表写入到名为output.csv
的CSV文件中。
下一篇:编写一个将字符串分割的函数