📅  最后修改于: 2023-12-03 15:33:58.813000             🧑  作者: Mango
如果你想要写入 CSV 数据,一个可行的方式是使用 Python 中的 csv
库。在 CSV 文件中,通常用逗号或分号分隔不同的数据字段。在 Python 中可以使用 csv.writer
或者 csv.DictWriter
这两个模块来写入 CSV 文件。
其中,csv.DictWriter
模块是使用 Python 字典类型写入 CSV 文件的一种方式。使用该模块时,你需要定义一个 Python 字典来代表每一行数据的内容,而数据字段可以直接使用字典的键值,这就免去了通过索引来确定数据字段的繁琐过程。这里有一个使用 csv.DictWriter
的简单例子:
import csv
with open('example.csv', 'w', newline='') as f:
fieldnames = ['name', 'age', 'city']
writer = csv.DictWriter(f, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'name': 'Alice', 'age': '25', 'city': 'New York'})
writer.writerow({'name': 'Bob', 'age': '30', 'city': 'San Francisco'})
writer.writerow({'name': 'Charlie', 'age': '35', 'city': 'Seattle'})
上述例子中,我们首先定义了一个字段名列表 fieldnames
,然后创建了一个 CSV 写入器 writer
,并指定字段名列表。然后通过使用 writeheader()
写入字段名头部,然后分别使用 writerow()
来写入每一行数据。这里我们使用了 Python 文本模式中的 'w'
模式创建 CSV 文件,同时也需要指定 newline=''
参数来规避换行符的差异问题。
csv.DictWriter
的另一个优点是可以提供参数来指定使用不同的分隔符或引号字符等等。详细参数可查看 Python doc 。