从python中的csv文件中提取列名

Question

我有一个包含100个列的大型csv文件。目前我能够读取csv文件及其特定行。我的文件在zipfile中，这是我到目前为止的代码。

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

next(csv.reader(f))将只返回第一行（可能是列名称所在的位置。）

Answer 2

只需按照您的代码获取文件的第一行：

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

希望有所帮助。