如何在csv fi中包含标题

Question

我有一个csv文件，不包括标题。如何手动添加标头，即将标头添加到文件顶部

GA, 11
LA, 12
NY, 122
......

Answer 1

正如Lutz Horn指出的，如果您使用的是py3.2+，则可以使用csv.writeheader。如果您像我一样使用py2.7，也许这会有所帮助：

import csv
myHeaders = ["Header1", "Header2"]
csvData = []
with open( 'myfile.csv','r+b' ) as csvfile:
    # Read the data
    reader = csv.reader(csvfile, quoting=csv.QUOTE_NONE)
    writer = csv.writer(csvfile)
    writer.writerow(myHeaders)
    writer.writerows([row for row in reader])

注意：csv没有insert/overwrite row方法，因此需要读写整个数据。对于中小型文件应该很简单

编辑：以前的代码会将重复的数据追加到文件的末尾