如何在python中向csv文件的特定列写入列表

import csv

def write_to_csv(filename, column_index, data):
    with open(filename, 'r') as file:
        reader = csv.reader(file)
        rows = list(reader)
    
    for row in rows:
        if column_index < len(row):
            row[column_index] = data.pop(0)
    
    with open(filename, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(rows)

# 示例用法
filename = 'data.csv'  # CSV文件名
column_index = 2  # 要写入的列索引（从0开始）
data = ['value1', 'value2', 'value3']  # 要写入的列表数据

write_to_csv(filename, column_index, data)