python 按列拆分csv

你可以使用 Python 自带的 csv 模块来读取和处理 CSV 文件，这里提供一种按列拆分 CSV 文件的方法：

import csv

# 读取 CSV 文件
with open('input.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    # 获取 CSV 表头
    headers = next(reader)
    # 获取需要拆分的列索引
    split_columns = [1, 3]  # 拆分第 1 列和第 3 列
    # 创建拆分后的 CSV 文件
    split_files = {}
    for col in split_columns:
        split_files[col] = open(f'output_col{col}.csv', 'w', newline='')
        writer = csv.writer(split_files[col])
        writer.writerow(headers)
    # 拆分 CSV 文件
    for row in reader:
        for col in split_columns:
            split_files[col].write(','.join([row[i] for i in range(len(headers)) if i == col]) + '\n')
    # 关闭拆分后的 CSV 文件
    for file in split_files.values():
        file.close()

上述代码将 `input.csv` 文件按照第 1 列和第 3 列进行拆分，并生成两个文件 `output_col1.csv` 和 `output_col3.csv`。你可以根据自己的需求修改拆分的列索引和拆分后的文件名。