用Python从CSV文件中提取列名

Question

我有一个很大的csv文件，里面有几百列。目前我可以读取这个csv文件和它的某一行。我的文件在一个压缩包里，这里是我现在写的代码。

import os, sys, csv, zipfile

zip_file = zipfile.ZipFile('some_zip_file.zip')
f = zip_file.open('some_csv_file.csv', 'r')

for row in csv.reader(f):
    print row[1]

f.close()

但是有没有办法只提取出csv文件的列名呢？我在用python 2.7。

Answer 1

只需用你的代码获取文件的第一行：

import os, sys, csv, zipfile

zip_file = zipfile.ZipFile('some_zip_file.zip')
f = zip_file.open('some_csv_file.csv', 'r')

for column in csv.reader(f).next():
    print column
    # collumn_name

希望这能帮到你。

Answer 2

next(csv.reader(f)) 这个代码会返回文件中的第一行，通常这一行是列名，也就是表格的标题。