python - 在 csv 文件中单独读取列名

Question

我有一个包含以下列的 csv 文件:

身份证、姓名、年龄、性别

后面是上面列的很多值。我正在尝试单独读取列名并将它们放入列表中。

我正在使用 Dictreader，这给出了正确的详细信息:

with open('details.csv') as csvfile:
    i=["name","age","sex"]
    re=csv.DictReader(csvfile)
    for row in re:
        for x in i:
            print row[x]

但我想要做的是，我需要使用输入 csv 自动解析列列表(在上述情况下为“i”)，而不是将它们硬编码在列表中。

with open('details.csv') as csvfile:
   
    rows=iter(csv.reader(csvfile)).next()
    header=rows[1:]
    re=csv.DictReader(csvfile)
    for row in re:
        print row
        for x in header:
            
            print row[x]

这会报错

Keyerrror:'name'

在行打印行[x]。我哪里错了？是否可以使用 Dictreader 获取列名？

Answer 1

虽然您已经有一个可接受的答案，但我想我会为其他对不同解决方案感兴趣的人添加此答案-

• CSV 模块中的 Python 的 DictReader 对象(从 Python 2.6 及更高版本开始)有一个名为 fieldnames 的公共(public)属性。 https://docs.python.org/3.4/library/csv.html#csv.csvreader.fieldnames

一个实现可能如下:

import csv

with open('C:/mypath/to/csvfile.csv', 'r') as f:
    d_reader = csv.DictReader(f)

    #get fieldnames from DictReader object and store in list
    headers = d_reader.fieldnames

    for line in d_reader:
        #print value in MyCol1 for each row
        print(line['MyCol1'])

在上面，d_reader.fieldnames 返回标题列表(假设标题在顶行)。这允许...

>>> print(headers)
['MyCol1', 'MyCol2', 'MyCol3']

如果您的标题在第 2 行(最上面的行是第 1 行)，您可以执行以下操作:

import csv

with open('C:/mypath/to/csvfile.csv', 'r') as f:
    #you can eat the first line before creating DictReader.
    #if no "fieldnames" param is passed into
    #DictReader object upon creation, DictReader
    #will read the upper-most line as the headers
    f.readline()

    d_reader = csv.DictReader(f)
    headers = d_reader.fieldnames

    for line in d_reader:
        #print value in MyCol1 for each row
        print(line['MyCol1'])