在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)