如何将标头包含到csv文件中

Question

I have a csv file that doesn't include headers. How can I manually add the headers ie add the headers to the top of the file

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

Answer 1

As Lutz Horn pointed out, you can use csv.writeheader if you're using py3.2+. If you're using py2.7 like me, maybe this might help:

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

Note : csv doesn't have an insert/overwrite row method, so you need to read & write the whole data. Should be trivial for small-to-medium sized files

Edit: Previous code would append repeated data to the end of the file