Python按行写入CSV

Python按行写入CSV

在实际开发中，我们经常需要将数据写入CSV文件中。Python的CSV模块提供了写入CSV文件的功能，可以按行写入数据。本文将详细介绍如何使用Python按行写入CSV文件。

1. 导入CSV模块

首先，我们需要导入Python的CSV模块。

import csv

2. 打开CSV文件

接下来，我们需要打开一个CSV文件并创建一个csv.writer对象。

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)

上面的代码中，我们打开一个名为output.csv的文件，并以写入模式打开。newline=''参数是为了避免在Windows系统下出现空行的问题。

3. 按行写入数据

接下来，我们可以按行写入数据到CSV文件中。

data = [
    ['Name', 'Age', 'City'],
    ['Alice', 25, 'New York'],
    ['Bob', 30, 'San Francisco'],
    ['Charlie', 35, 'Chicago']
]

for row in data:
    writer.writerow(row)

上面的代码定义了一个二维列表data，然后遍历这个列表，每次写入一行数据到CSV文件中。writer.writerow()方法用于写入一行数据。

4. 完整示例

下面是一个完整的示例代码，演示了如何按行写入数据到CSV文件中。

import csv

data = [
    ['Name', 'Age', 'City'],
    ['Alice', 25, 'New York'],
    ['Bob', 30, 'San Francisco'],
    ['Charlie', 35, 'Chicago']
]

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    for row in data:
        writer.writerow(row)

print('数据写入完成！')

5. 运行结果

运行上面的示例代码，会生成一个名为output.csv的CSV文件，并将数据按行写入到文件中。

Name,Age,City
Alice,25,New York
Bob,30,San Francisco
Charlie,35,Chicago

总结

本文介绍了如何使用Python的CSV模块按行写入数据到CSV文件中。首先导入CSV模块，然后打开CSV文件并创建一个csv.writer对象，接着按行写入数据到文件中。