Python写入csv并添加新行

有时候我们需要将数据写入到csv文件中并且在已有的内容之后添加新的行。Python内置的csv模块提供了很好的支持。本文将介绍如何使用Python写入csv文件并添加新行。

1. csv模块

首先需要导入csv模块：

import csv

2. 写入csv文件

要写入csv文件，我们需要打开一个文件，使用csv.writer()创建一个csv写入器并用writerow()方法写入数据。例如：

with open('data.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age'])
    writer.writerow(['Alice', 25])
    writer.writerow(['Bob', 30])

第一个参数是文件名，第二个参数mode='w'表示写入模式，而newline=''表示在写入文件的时候不添加额外的空行。

这个代码片段创建了一个名为'data.csv'的文件，并写入了三行数据。第一行是表头，后面两行是数据。现在数据是紧凑的，因为没有添加额外的空行。如果需要在每行数据之间添加空行，可以将newline参数设置为'\n'。

3. 添加新行

要向已有的csv文件中添加新行，需要打开文件为追加模式，并创建csv写入器。例如：

with open('data.csv', mode='a', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Charlie', 35])

第二个参数mode='a'表示追加模式，因此这个代码片段将新数据添加到现有文件的末尾。

4. 完整的代码

以下是完整的代码，其中示例数据被写入到'data.csv'文件中，并添加了一个新行：

import csv

with open('data.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age'])
    writer.writerow(['Alice', 25])
    writer.writerow(['Bob', 30])

with open('data.csv', mode='a', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Charlie', 35])

这个代码片段执行后，'data.csv'文件内容如下：

Name,Age
Alice,25
Bob,30
Charlie,35

结论

使用Python，我们可以轻松地将数据写入到csv文件中并添加新行。csv模块为数据管理提供了很好的支持。