Python - 将列表字典写入 CSV

在数据处理中，CSV（逗号分隔值）文件格式是常见的数据交换格式。在Python中，我们可以使用CSV模块来读取和写入CSV文件。本文将介绍如何将Python列表字典写入CSV文件。

准备工作

在开始之前，我们需要先创建一个列表字典。在本例中，我们将创建一个将包含员工信息的Python列表字典。该字典将包含以下键值对：

employee_list = [
  {"Name": "John", "Age": 28, "Department": "Marketing", "Salary": 40000},
  {"Name": "Sarah", "Age": 25, "Department": "Sales", "Salary": 35000},
  {"Name": "Tom", "Age": 32, "Department": "Accounting", "Salary": 50000},
  {"Name": "Emily", "Age": 24, "Department": "IT", "Salary": 45000}
]

写入CSV文件

要将上述Python列表字典写入CSV文件，我们需要使用CSV模块中的writer()函数。以下是写入CSV文件的代码片段：

import csv

with open('employee_data.csv', mode='w', newline='') as employee_file:
    fieldnames = ['Name', 'Age', 'Department', 'Salary']
    writer = csv.DictWriter(employee_file, fieldnames=fieldnames)
    
    writer.writeheader()
    for employee in employee_list:
        writer.writerow(employee)

这段代码创建了一个名为employee_data.csv的新CSV文件。我们打开文件并使用csv.DictWriter类创建一个字典写入器对象。我们要向文件中写入的每一行都需要提供一个字典，将每个键映射到CSV文件中的相应列。

在第6行代码中，writeheader()方法将在CSV文件的首行写入我们在第3行代码中定义的字段名称。在下一个代码行中，我们遍历employee_list中的每个员工字典，并将字典传给writerow()方法以将其写入CSV文件中。

最终结果是创建了一个名为employee_data.csv的文件，并将包含在employee_list中的员工信息写入了文件。文件内容如下：

Name,Age,Department,Salary
John,28,Marketing,40000
Sarah,25,Sales,35000
Tom,32,Accounting,50000
Emily,24,IT,45000

总结

如上所述，使用Python的CSV模块，我们可以轻松地将列表字典写入CSV文件。这使得我们可以轻松地在Python中处理和交换数据。