python中csv模块中DictWriter()函数的作用?所需参数?简单案例?
时间: 2024-05-12 12:18:26 浏览: 27
好的,我了解了您的要求,以下是您的问题的答案:
csv模块中DictWriter()函数的作用是将字典数据写入CSV文件中。它可以将字典对象转换成CSV文件中的一行数据,并且可以有选择性地包含或排除特定的键值对。
DictWriter()函数需要两个参数,一个是文件对象,用于写入CSV数据,另一个是包含要写入的字段名称的列表。
以下是一个简单的例子,演示了如何使用csv模块中的DictWriter()函数将字典对象写入CSV文档中:
```
import csv
# 要写入的数据字典
data = [{'name': 'John', 'age': 25, 'location': 'New York'},
{'name': 'Linda', 'age': 30, 'location': 'Chicago'},
{'name': 'Mike', 'age': 35, 'location': 'Los Angeles'}]
# 将数据写入CSV文件中
with open('people.csv', 'w', newline='') as f:
# 指定字段名称
fieldnames = ['name', 'age', 'location']
# 创建DictWriter对象
writer = csv.DictWriter(f, fieldnames=fieldnames)
# 写入文件头
writer.writeheader()
# 写入数据
for person in data:
writer.writerow(person)
```
在这个例子中,当我们运行这个脚本时,它将创建一个名为“people.csv”的新CSV文件,并将数据写入其中。如果您打开这个文件,您将看到以下内容:
```
name,age,location
John,25,New York
Linda,30,Chicago
Mike,35,Los Angeles
```
希望这可以帮助您回答您的问题!