Python 实现数据写入 CSV 文件

在 Python 编程中，将数据写入 CSV（逗号分隔值）文件是一项常见且重要的任务。CSV 文件格式简单易用，被广泛用于数据存储和交换。下面我们将详细介绍如何使用 Python 实现数据写入 CSV 文件。

我们需要导入 Python 内置的csv模块。

import csv

接下来，假设我们有一些数据需要写入 CSV 文件。例如，我们有一个包含学生信息的列表：

student_data = [
    ['张三', 20, '计算机科学'],
    ['李四', 21, '电子工程'],
    ['王五', 19, '数学']
]

要将这些数据写入 CSV 文件，我们可以使用以下代码：

with open('students.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(student_data)

在上述代码中，open('students.csv', 'w', newline='')以写入模式打开一个名为students.csv的文件。csv.writer对象用于将数据写入 CSV 文件。writerows方法接受一个可迭代对象（如列表），并将其内容逐行写入 CSV 文件。

如果我们希望在 CSV 文件中添加标题行，可以这样做：

headers = ['姓名', '年龄', '专业']
with open('students.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(headers)
    writer.writerows(student_data)

我们还可以对写入的数据进行更多的控制。例如，指定分隔符（默认是逗号）、设置引号规则等。

with open('students.csv', 'w', newline='') as file:
    writer = csv.writer(file, delimiter=';', quotechar='"')
    writer.writerow(headers)
    writer.writerows(student_data)

在实际应用中，我们可能从数据库、网络请求或其他数据源获取数据，并将其写入 CSV 文件。Python 的csv模块提供了简单而强大的功能，使这一过程变得轻松高效。

通过 Python 的csv模块，我们能够方便地将各种数据以 CSV 格式进行存储和输出，为数据处理和分析提供了极大的便利。无论是处理小规模的本地数据，还是与其他系统进行数据交换，掌握 Python 写入 CSV 文件的方法都是非常有用的。