python在csv中添加一列数据

可以使用Python内置的csv模块来读取csv文件并添加一列数据。

假设我们有一个名为example.csv的文件，其中包含以下数据：

Name,Age,Gender
Alice,25,Female
Bob,30,Male

现在我们想要添加一列"Occupation"，可以按照以下步骤进行操作：

import csv

# 1. 读取csv文件并获取原始数据
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    data = [row for row in reader]

# 2. 添加新的一列数据
header = data[0]  # 获取表头
header.append('Occupation')  # 添加新的表头
for row in data[1:]:  # 遍历每一行数据并添加新的一列
    if row[0] == 'Alice':
        row.append('Engineer')
    elif row[0] == 'Bob':
        row.append('Manager')

# 3. 将新的数据写入csv文件
with open('example.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

执行上述代码后，example.csv文件将变为：

Name,Age,Gender,Occupation
Alice,25,Female,Engineer
Bob,30,Male,Manager

可以看到，新的一列数据"Occupation"已经成功添加到了csv文件中。