csv.DictWriter()
时间: 2024-09-09 21:00:48 浏览: 65
csv.DictWriter() 是Python标准库中csv模块的一个类,它专门用于将字典写入到CSV文件中。这个类可以帮助我们方便地处理字典数据,并将其写入CSV文件,其中字典的键作为CSV文件的列标题,字典的值则作为数据填充到相应的列中。
使用csv.DictWriter()时,通常需要执行以下几个步骤:
1. 创建一个csv.DictWriter实例,传入文件对象和字段名列表,字段名列表指定了CSV文件的列标题。
2. 可以选择性地调用writeheader()方法来写入列标题。
3. 使用writerow()方法写入字典,其中字典的键应该是与字段名列表匹配的。
4. 如果需要,可以多次调用writerow()方法写入多行数据。
下面是一个使用csv.DictWriter()的基本示例:
```python
import csv
# 假设我们要写入的数据是以下字典的集合
data = [
{'name': 'Alice', 'age': 24},
{'name': 'Bob', 'age': 26},
{'name': 'Charlie', 'age': 28}
]
# 打开一个文件用于写入CSV数据
with open('output.csv', 'w', newline='') as csvfile:
# 创建DictWriter实例,指定文件对象和字段名列表
fieldnames = ['name', 'age']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
# 写入列标题
writer.writeheader()
# 写入数据行
for row in data:
writer.writerow(row)
```
以上代码将生成一个CSV文件,内容包含两列:'name' 和 'age',并且包含三行数据,对应于提供的字典集合。
阅读全文