python - Python将标题添加到CSV

Question

我正在使用 Django，但我认为这是一个 Python 问题....

如何在已上传的 CSV 文件顶部修改和写入一些标题？

文件上传

jack;1st street ;New York

我要附加到 CSV 的内容

name;street;area
jack;1st street ;New York

这是我上传后的阅读方式：

 import csv
    file = upload.filepath
    file_read = csv.reader(file.read().splitlines())

我怎样才能做到这一点？

Answer 1

您可以使用 DictReader 和 DictWriter 保存为带有标题的新文件...

file_read = csv.DictReader(open('/tmp/test.csv', 'rb'), 
    ['name','street','area'], delimiter=';')
file_write = csv.DictWriter(open('/tmp/test-headers.csv', 'wb'), 
    ['name','street','area'], delimiter=';')
file_write.writeheader()
file_write.writerows(file_read)

或者，如果您只想在脚本中使用该文件：

file_read = csv.DictReader(open('/tmp/test.csv', 'rb'), 
    ['name','street','area'], delimiter=';')

for row in file_read:
    print row

以上打印：

{'street': ' 1st street', 'name': 'jack', 'area': ' New York'}
{'street': ' 2nd street', 'name': 'sally', 'area': ' New York'}