遍历CSV时如何避免列名行被消耗？解决输出CSV列名缺失问题

你的代码循环逻辑确实存在问题，直接导致了列名行丢失：

1. 列名行未写入输出文件：循环中，当遇到第一个含逗号的行（也就是col1,col2,col3列名行）时，因为此时extra已经被设为True（前面的表头行都不含逗号），触发elif extra: break直接跳出循环，这一行根本没被写入output_file.csv。
2. pandas未读取到列名行：Python的文件迭代器for line in f1是逐行读取的，当line变量拿到列名行时，文件指针已经移动到下一行（数据行第一行）。后续pd.read_csv(f1)从数据行开始读，无法将列名行识别为DataFrame的列名，只能用默认的数字列名。
3. 最终输出无列名：虽然df.to_csv()默认会写入列名，但此时的列名是数字而非原列名，最终输出就缺失了你需要保留的原列名行。

调整循环逻辑，确保列名行被写入输出文件，同时让pandas正确读取列名：

import pandas as pd

with open('input_file.csv', 'r') as f1, open('output_file.csv', 'w') as f2:
    extra = False
    for line in f1:
        if ',' not in line:
            extra = True
            f2.write(line)
        else:
            # 写入列名行到输出文件
            f2.write(line)
            break

    # 从当前指针位置读取数据，pandas会自动将已写入的列名行作为表头
    df = pd.read_csv(f1)

    # 基础数据处理
    df.iat[0, 1] = 0 

    # 写入处理后的数据，跳过列名（已经手动写入过）
    df.to_csv(f2, index=False, header=False)

• 遇到含逗号的列名行时，先写入输出文件再break，保留原列名行。
• 调用to_csv时添加header=False，避免重复写入列名（因为已经手动写入过一次）。

内容的提问来源于stack exchange，提问作者user21474411