You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

字典转CSV追加写入时重复写入表头的问题求助

解决CSV追加时重复写入表头的问题

问题根源是每次调用is_known()函数时,都会执行writer.writeheader(),导致每次追加数据时重复写入表头。只需通过判断文件状态,仅在文件首次创建时写入表头即可解决。

修改后的代码

import os
import csv

def is_known():
    file_path = "data/data_known.csv"
    # 检查文件是否存在且已有内容
    file_has_content = os.path.exists(file_path) and os.path.getsize(file_path) > 0
    
    with open(file_path, "a", newline="") as f:
        writer = csv.DictWriter(f, fieldnames=current_card.keys())
        # 仅当文件不存在或为空时写入表头
        if not file_has_content:
            writer.writeheader()
        writer.writerow(current_card)

关键说明

  1. 借助os模块判断文件状态:
    • os.path.exists(file_path):确认文件是否已创建
    • os.path.getsize(file_path) > 0:避免文件存在但无内容时重复写入表头
  2. 仅在文件首次创建(或为空)时写入表头,后续追加操作仅写入数据行。

内容的提问来源于stack exchange,提问作者Mayur

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.25 22:37:39