从csv中选择特定列并更新另一个csv

import csv

def select_columns(source_csv, target_csv, columns):
    # 读取源csv文件
    with open(source_csv, 'r') as file:
        reader = csv.reader(file)
        data = list(reader)

    # 创建新的csv文件
    with open(target_csv, 'w', newline='') as file:
        writer = csv.writer(file)

        # 写入特定列数据
        for row in data:
            selected_row = [row[column] for column in columns]
            writer.writerow(selected_row)

    print("特定列已成功写入新的csv文件。")

# 示例用法
source_csv = 'source.csv'
target_csv = 'target.csv'
columns = [0, 2]  # 选择第1列和第3列

select_columns(source_csv, target_csv, columns)