在CSV-fi中写入标题

Question

我有一个程序，它将在cron作业上运行，并将输出写入CSV文件。你知道吗

我得到它写正确，但我想让程序写头的第一行时，文件被创建。程序是否有方法检查CSV文件中是否有行，如果没有，则写入标题。你知道吗

Answer 1

只需使用一个标志：

headers_written = False

然后在写入行时：

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

writer.writerow(somerow)

实际上，你应该推迟创建writer，直到你确定你有东西要写：

writer = None

# ...

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

writer.writerow(somerow)