Python 写入 CSV

CSV 是一种常用的电子表格数据格式，Python 中有很多库可以用来读写 CSV 文件。本文将介绍如何使用 Python 将数据写入 CSV 文件中。

1. 写入简单文本数据

使用 csv 库可以很容易地将文本数据写入 CSV 文件。

# 引入 csv 库
import csv

# 写入文件的数据
data = [
    ['姓名', '年龄', '性别'],
    ['张三', '20', '男'],
    ['李四', '22', '女'],
    ['王五', '25', '男']
]

# 打开 CSV 文件，开始写入
with open('data.csv', 'w', newline='', encoding='utf-8') as csvfile:
    writer = csv.writer(csvfile)
    for row in data:
        writer.writerow(row)

在上述代码中，我们创建了一个包含若干列表的列表 data，每一个列表代表一行数据。然后，使用 csv.writer() 方法实例化了一个 CSV 文件的写入器，并将每一行数据写入到文件中。

在 with 语句块内部，我们使用 open() 方法打开一个 CSV 文件，参数 newline='' 是为了避免出现空行；encoding='utf-8' 则是用来指定文本编码。最后，使用 writerows() 方法将整个二维列表写入到文件中。

2. 写入字典数据

如果要将字典数据写入到 CSV 文件中，可以使用 csv.DictWriter() 方法。该方法接受一个字典作为数据源，并将其转换为二维列表格式进行写入。

# 引入 csv 库
import csv

# 写入文件的数据
data = [
    {'姓名': '张三', '年龄': '20', '性别': '男'},
    {'姓名': '李四', '年龄': '22', '性别': '女'},
    {'姓名': '王五', '年龄': '25', '性别': '男'}
]

# 打开 CSV 文件，开始写入
with open('data.csv', 'w', newline='', encoding='utf-8') as csvfile:
    writer = csv.DictWriter(csvfile, fieldnames=data[0].keys())
    writer.writeheader()
    writer.writerows(data)

在上述代码中，我们先创建了一个列表 data，其中包含三个字典。然后，使用 csv.DictWriter() 方法实例化了一个 CSV 文件的写入器，并使用 fieldnames 参数来指定列名，这里使用了数据源的第一个字典的 keys() 方法获取。

使用 writeheader() 方法可以将列名写入到文件中，如果不需要列名，则可以省略该方法。最后，使用 writerows() 方法将整个列表写入到文件中。

参考资料

• Python CSV Module Documentation