字典转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)
关键说明
- 借助
os模块判断文件状态:os.path.exists(file_path):确认文件是否已创建os.path.getsize(file_path) > 0:避免文件存在但无内容时重复写入表头
- 仅在文件首次创建(或为空)时写入表头,后续追加操作仅写入数据行。
内容的提问来源于stack exchange,提问作者Mayur
相关产品推荐
相关产品推荐

