在CSV文件中编写标题

Question

我有一个程序将在cron作业上运行并将输出写入CSV文件。

我让它写得正确，但我想在创建文件时让程序在第一行写入标题。是否有一种方法可以检查CSV文件中是否有任何行，如果没有，则写入标题。

Answer 1

只需使用标志：

headers_written = False

然后写行：

if not headers_written:
    writer.writerow(headers)
    headers_written = True

writer.writerow(somerow)

你确实推迟创作作家，直到你确定要写东西为止：

writer = None

# ...

if not writer:
    writer = csv.writer(open(filename, 'wb'))
    writer.writerow(headers)

writer.writerow(somerow)