Python中在csv中添加新列并写入数据的实现方法
简介
在Python中，我们经常会遇到需要在csv文件中添加新列并写入数据的情况。本文将介绍如何使用Python实现这一功能。
流程图
flowchart TD
A[读取CSV文件] --> B[添加新列]
B --> C[写入数据]
C --> D[保存CSV文件]
代码实现步骤
1. 导入必要的模块
首先，我们需要导入csv模块，以便进行CSV文件的读取、写入和操作。
import csv
2. 读取CSV文件
使用csv.reader函数来读取CSV文件。该函数会返回一个迭代器，我们可以通过for循环逐行读取文件。
with open('input.csv', 'r') as file:
reader = csv.reader(file)
data = list(reader)
3. 添加新列
在CSV文件中添加新列，需要先找到每一行的位置，在该位置插入新列的值。
# 在表头添加新列的名称，例如'new_column'
data[0].append('new_column')
# 在每一行的相应位置插入新列的值
for row in data[1:]:
row.append('new_value')
4. 写入数据
使用csv.writer函数来写入修改后的数据到CSV文件中。
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
5. 保存CSV文件
最后，使用with语句打开一个新的文件，将修改后的数据写入到该文件，并保存。
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
代码说明
下面是对代码中关键部分的注释说明：
# 读取CSV文件
with open('input.csv', 'r') as file:
reader = csv.reader(file)
data = list(reader)
在这段代码中，我们使用了with语句来打开CSV文件，并将文件内容读取到data列表中。
# 添加新列的名称
data[0].append('new_column')
# 在每一行的相应位置插入新列的值
for row in data[1:]:
row.append('new_value')
这段代码中，我们在第一行的末尾添加了新列的名称，并在每一行的末尾插入了新列的值。
# 写入数据到CSV文件
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
在这段代码中，我们使用with语句打开一个新的文件，并使用csv.writer函数将修改后的数据写入到该文件中。writerows方法用于将整个二维列表（data）写入到文件中。
总结
通过以上步骤，我们可以实现在Python中在CSV文件中添加新列并写入数据的功能。首先，我们使用csv.reader函数读取CSV文件的内容，并将其保存在一个二维列表中。然后，我们在表头添加新列的名称，并在每一行的末尾插入新列的值。最后，使用csv.writer函数将修改后的数据写入到另一个文件中，完成整个过程。
希望本文对你有所帮助，祝你在Python开发的道路上越来越进步！
