CSV文件按列前缀拆分的代码错误排查及表头写入问题问询

我来帮你精准找出代码里的两个关键问题，这就是导致输出文件内容混乱、甚至触发StopIteration错误的根源：

第一个致命错误：多次调用next(cr)消耗了不该消耗的行 #

csv.reader返回的是一个迭代器，每次调用next(cr)都会读取原文件的下一行。你现在连续调用了四次next(cr)：

• 第一次调用确实拿到了原文件的表头，写入了column1_01.csv
• 第二次调用读取的是原文件的第一行数据（比如你示例里的01;BE），却被当成表头写入了column1_02.csv
• 第三次调用读取第二行数据02;ED，当成表头写入column1_03.csv
• 第四次调用读取第三行数据12;FD，当成表头写入column1_04.csv

这直接导致三个问题：

1. 所有输出文件的“表头”全是错的，根本不是原文件的表头
2. 原文件的前3行数据被白白浪费，没有进入后续的分类逻辑
3. 如果原文件数据量很小（比如测试样本），迭代器很快就会被耗尽，再调用next(cr)就会触发StopIteration错误，这就是你遇到的报错原因

第二个问题：表头没有统一复用 #

你想要每个输出文件都使用原文件的同一个表头，但当前写法是每次next(cr)取新行，完全没做到复用，反而把数据行当成了表头。

修正后的代码（完全符合你的要求：纯csv、无额外函数/异常处理） #

import csv
import os  # 补上你遗漏的os模块导入，不然os.path会报错

output_path=r'C:\myfolder\large_file.csv'

with open(os.path.join(os.path.dirname(output_path),"column1_01.csv"), "w", encoding="utf-8", newline='') as out_01, \
     open(os.path.join(os.path.dirname(output_path),"column1_02.csv"), "w", encoding="utf-8", newline='') as out_02, \
     open(os.path.join(os.path.dirname(output_path),"column1_03.csv"), "w", encoding="utf-8", newline='') as out_03, \
     open(os.path.join(os.path.dirname(output_path),"column1_04.csv"), "w", encoding="utf-8", newline='') as out_04:

    cw01 = csv.writer(out_01, delimiter=";", quoting=csv.QUOTE_MINIMAL)
    cw02 = csv.writer(out_02, delimiter=";", quoting=csv.QUOTE_MINIMAL)
    cw03 = csv.writer(out_03, delimiter=";", quoting=csv.QUOTE_MINIMAL)
    cw04 = csv.writer(out_04, delimiter=";", quoting=csv.QUOTE_MINIMAL)

    with open(output_path, encoding="utf-8") as in_f:
        cr = csv.reader(in_f, delimiter=";")
        # 只调用一次next(cr)，拿到真正的表头并保存
        header = next(cr)
        # 将同一个表头写入所有输出文件
        cw01.writerow(header)
        cw02.writerow(header)
        cw03.writerow(header)
        cw04.writerow(header)

        # 遍历剩下的所有数据行，按前缀分类写入
        for line in cr:
            prefix = line[0][:2]
            if prefix == "01":
                cw01.writerow(line)
            elif prefix == "02":
                cw02.writerow(line)
            elif prefix == "03":
                cw03.writerow(line)
            elif prefix == "04":
                cw04.writerow(line)

关键修正点说明： #

1. 补上了遗漏的import os：你原来的代码里使用了os.path但没导入模块，这是一个隐藏的运行错误
2. 只调用一次next(cr)获取表头：把表头保存到变量header，然后统一写入所有输出文件，保证表头完全一致
3. 把多个独立if改成elif：虽然不是必须，但能减少不必要的判断，效率更高（匹配到一个条件后就不会再检查后续条件）

备注：内容来源于stack exchange，提问作者PSt