Python csv：优化标头输出

Question

我正在将大量 (160k+) 数据从 SQL 写入 CSV。我的脚本功能完全符合预期，但我确信必须有一种更有效的方式在输出中包含标题。我通过阅读 writing header in csv python with DictWriter 拼凑出以下内容，但觉得它缺乏优雅。

这是我的代码：

f = open(outfile,'w')
wf = csv.DictWriter(f, fieldnames, restval='OOPS')
wf.writer.writerow(wf.fieldnames)
f.close()

f = open(outfile,'a')
wf = csv.writer(f)
wf.writerows(rows)
f.close()

fieldnames 已明确定义（10 个自定义列名），rows 包含来自我的查询的 fetchall()。

Answer 1

未经测试，但我不明白为什么这不应该做这项工作：

import csv

with open(outfile, "wb") as outf:
    outcsv = csv.writer(outf)
    outcsv.writerow(fieldnames)
    outcsv.writerows(rows)