python - 使用字典检索 csv 的第一行

Question

我正在使用 csv.DictReader 类读取一个 csv 文件。我在 python 文档中读到 class csv.DictReader(csvfile, fieldnames=None, restkey=None, restval=None, dialect='excel', *args, **kwds)

如果省略 fieldnames 参数，csv 文件第一行的值将用作字段名。

我尝试使用字典的 keys() 方法获取我的 csv 文件的第一行:

my_reader = csv.DictReader(src)
print(my_reader.keys())

这样做，我得到以下错误:

    print(my_reader.keys())
    AttributeError: 'DictReader' object has no attribute 'keys'

为什么？

Answer 1

字段名存储在属性 fieldnames 中:

my_reader = csv.DictReader(src)
print(my_reader.fieldnames)

keys 是一种字典方法。 my_reader 不是字典。但是，它是一个生成字典的迭代器，因此您也可以这样做:

for row in my_reader:
    print(row.keys())