从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

希望对您有所帮助。