对csv文件中的列进行重新排序

import csv

def reorder_csv_columns(csv_file, column_order):
    # 读取CSV文件
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        rows = list(reader)

    # 获取列名
    header = rows[0]

    # 确定列的索引顺序
    column_indices = [header.index(column) for column in column_order]

    # 重新排序列
    reordered_rows = []
    for row in rows:
        reordered_row = [row[index] for index in column_indices]
        reordered_rows.append(reordered_row)

    # 更新CSV数据
    rows = reordered_rows

    # 写入CSV文件
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(rows)

# 示例用法
csv_file = 'data.csv'
column_order = ['Column3', 'Column1', 'Column2']  # 以列表形式指定列的顺序
reorder_csv_columns(csv_file, column_order)