向csv文件中的指定索引处添加具有不同值的列

import csv

def add_column_to_csv(filename, index, column_values):
    # 打开CSV文件
    with open(filename, 'r') as file:
        # 读取CSV文件内容
        reader = csv.reader(file)
        rows = list(reader)
    
    # 确认索引是否在有效范围内
    if index < 0 or index > len(rows[0]):
        print("指定索引超出范围")
        return
    
    # 在指定索引处添加列标题
    rows[0].insert(index, "New Column")
    
    # 生成不同的值用于新列
    new_column_values = column_values
    
    # 逐行添加新列的值
    for i in range(1, len(rows)):
        new_value = new_column_values[i - 1]
        rows[i].insert(index, new_value)
    
    # 将更新后的数据写入CSV文件
    with open(filename, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(rows)

# 示例用法
filename = 'data.csv'
index = 2
column_values = ['Value 1', 'Value 2', 'Value 3', 'Value 4']

add_column_to_csv(filename, index, column_values)