Python创建追加CSV文件时如何仅添加一次表头？

我来帮你搞定这个重复添加表头的问题！你的代码核心问题在于每次调用handle_data函数时，都会无条件写入表头，这就导致每批数据前都重复出现列名。下面给你两种简洁的解决方案：

方法一：用Pandas自带的to_csv（推荐） #

既然你已经在用Pandas的DataFrame，完全可以利用它的to_csv方法来自动处理表头逻辑，比手动写CSV更省心：

修改handle_data函数里的文件写入部分，替换掉原来的csv.writer代码：

import os  # 需要导入os模块判断文件状态

def handle_data(data):
    msg5 = data.replace("\n"," ").replace("\r"," ").split(",")
    print(msg5)
    d = dict(s.split(':') for s in msg5)
    data_frame = pd.DataFrame(list(d.items())).transpose()
    data_frame.columns = data_frame.iloc[0]
    data_frame = data_frame.reindex(data_frame.index.drop(0))
    print(data_frame)
    
    filename = value_text + time2 + ".csv"
    # 检查文件是否已存在
    file_exists = os.path.isfile(filename)
    # 追加写入：文件不存在时写表头，否则只写数据；index=False避免写入索引列
    data_frame.to_csv(filename, mode='a', header=not file_exists, index=False)

原理说明： #

• mode='a'：以追加模式打开文件
• header=not file_exists：只有当文件是新创建的（不存在）时，才写入表头
• index=False：避免把DataFrame的索引列写入CSV（如果不需要索引的话）

方法二：手动用csv模块控制表头写入 #

如果你想继续用csv模块，只需要在写入表头前判断文件是否为空即可：

import os  # 导入os模块

def handle_data(data):
    # ... 前面的DataFrame处理代码不变 ...
    
    filename = value_text + time2 + ".csv"
    fields = list(data_frame.columns)
    
    with open(filename, 'a', newline='') as writeFile:
        writeFile = csv.writer(writeFile)
        # 检查文件大小，为0说明是新文件，写入表头
        if os.path.getsize(filename) == 0:
            writeFile.writerow(fields)
        writeFile.writerows(data_frame.values)

原理说明： #

• os.path.getsize(filename) == 0：判断文件是否为空（新创建的文件大小为0），只有满足条件时才写入表头
• 后续追加数据时，因为文件已有内容，不会重复写表头

两种方法都能实现你想要的效果：只在第一次写入时添加表头，后续只追加数据行。

内容的提问来源于stack exchange，提问作者Rahul R Kulkarni