Python循环写入CSV文件时如何仅保留一次表头?
解决CSV重复写入表头的问题
核心思路:仅在文件不存在或为空时写入表头,后续追加数据时跳过表头写入。
方式1:检查文件是否存在
通过os.path.exists判断文件是否存在,首次运行时文件不存在,写入表头;后续运行直接追加数据:
import csv import os # 假设now和Temperatura已提前获取 file_path = 'Temp.csv' file_exists = os.path.exists(file_path) with open(file_path, 'a', newline='') as file: fieldnames = ['Date', 'Temp'] writer = csv.DictWriter(file, fieldnames=fieldnames) # 仅文件不存在时写入表头 if not file_exists: writer.writeheader() writer.writerow({'Date': now, 'Temp': Temperatura})
方式2:检查文件是否存在且非空(更严谨)
如果存在空文件(比如手动创建了空的Temp.csv),方式1会跳过表头导致文件无表头。这种情况可以通过os.path.getsize检查文件大小,确保空文件也能写入表头:
import csv import os file_path = 'Temp.csv' # 判断文件存在且大小大于0(即已有内容) file_has_content = os.path.exists(file_path) and os.path.getsize(file_path) > 0 with open(file_path, 'a', newline='') as file: fieldnames = ['Date', 'Temp'] writer = csv.DictWriter(file, fieldnames=fieldnames) if not file_has_content: writer.writeheader() writer.writerow({'Date': now, 'Temp': Temperatura})
内容的提问来源于stack exchange,提问作者SzypkiWonsz
相关产品推荐
相关产品推荐

