📜  python csv dictwriter - Python (1)

📅  最后修改于: 2023-12-03 15:33:58.813000             🧑  作者: Mango

Python csv DictWriter

如果你想要写入 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