DictWriter 函数 python
时间: 2024-05-01 11:21:44 浏览: 105
`DictWriter` 是Python中的一个CSV文件写入函数,主要用于将字典数据写入到CSV文件中。
使用`DictWriter`函数,需要先导入`csv`模块。然后,创建一个CSV文件并打开,将要写入的字典数据作为参数传递给`DictWriter`函数,最后使用`writeheader()`方法写入表头,使用`writerows()`方法写入数据。
以下是一个示例代码:
```python
import csv
# 创建CSV文件并打开
with open('example.csv', mode='w', newline='') as csv_file:
fieldnames = ['name', 'age', 'gender']
writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
# 写入表头
writer.writeheader()
# 写入数据
writer.writerow({'name': 'Tom', 'age': 20, 'gender': 'male'})
writer.writerow({'name': 'Lucy', 'age': 18, 'gender': 'female'})
```
以上代码将创建一个名为`example.csv`的文件,并将表头`name`、`age`和`gender`写入文件中。接着,将字典数据写入到文件中,最终的CSV文件内容如下:
```
name,age,gender
Tom,20,male
Lucy,18,female
```
阅读全文