在csv文件的特定列中插入数据

import csv

def insert_data_in_csv(csv_file, column_name, data):
    # 读取csv文件
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        rows = list(reader)

    # 确定要插入数据的特定列
    column_index = reader.fieldnames.index(column_name)

    # 在特定列中插入数据
    for row in rows:
        row[column_name] = data

    # 将更新后的数据写入csv文件
    with open(csv_file, 'w', newline='') as file:
        writer = csv.DictWriter(file, fieldnames=reader.fieldnames)
        writer.writeheader()
        writer.writerows(rows)

# 示例用法
csv_file = 'data.csv'
column_name = '特定列'
data = '插入的数据'

insert_data_in_csv(csv_file, column_name, data)