Python写入CSV文件的方法

下面是关于Python写入CSV文件的方法的完整攻略。

什么是CSV文件？

CSV（Comma-Separated Values）是一种简单和常见的数据传输格式，也是许多数据工具和应用程序使用的一种标准格式。CSV文件由一系列以逗号分隔的行组成，每行代表一条记录，每个字段以逗号分隔，没有表头。

如何写入CSV文件？

Python提供了内置模块csv来读写CSV文件，可以使用该模块的writer对象将数据写入CSV文件。

1. 使用csv.writer写入CSV文件

下面是一个示例代码，说明如何使用csv.writer写入CSV文件。该代码将一组学生成绩写入CSV文件。

import csv

# 打开文件，以写入模式
with open('students.csv', mode='w', newline='') as file:
    # 创建writer对象
    writer = csv.writer(file)

    # 写入表头
    writer.writerow(['Name', 'Age', 'Score'])

    # 写入数据
    writer.writerow(['Tom', '18', '98'])
    writer.writerow(['Jerry', '19', '88'])
    writer.writerow(['Lucy', '17', '90'])

代码执行后，将生成一个名为students.csv的文件，内容如下：

Name,Age,Score
Tom,18,98
Jerry,19,88
Lucy,17,90

2. 使用csv.DictWriter写入CSV文件

如果需要写入带表头的CSV文件，可以使用csv.DictWriter对象。

下面是一个示例代码，说明如何使用csv.DictWriter写入CSV文件。该代码将一组学生成绩写入CSV文件。

import csv

# 定义表头
header = ['Name', 'Age', 'Score']

# 定义数据
data = [
    {'Name': 'Tom', 'Age': '18', 'Score': '98'},
    {'Name': 'Jerry', 'Age': '19', 'Score': '88'},
    {'Name': 'Lucy', 'Age': '17', 'Score': '90'},
]

# 打开文件，以写入模式
with open('students.csv', mode='w', newline='') as file:
    # 创建DictWriter对象，写入表头
    writer = csv.DictWriter(file, fieldnames=header)
    writer.writeheader()

    # 写入数据
    writer.writerows(data)

代码执行后，将生成一个名为students.csv的文件，内容如下：

Name,Age,Score
Tom,18,98
Jerry,19,88
Lucy,17,90

总结

以上就是Python写入CSV文件的方法。可以使用csv.writer和csv.DictWriter对象写入CSV文件，前者适用于没有表头的文件，而后者适用于带表头的文件。在使用csv模块时，务必注意编码和文件读写模式等问题。