python - 将 csv 文件的第二行读入字典中

Question

我有一个 csv 文件，看起来像这样:

username,permission,type,avtar
user1,op,member,goldilock
user2,guest
user3,guest

“type”和“avtar”数据仅位于第二行(user1)，对于 user2 和 user3，该数据将为空。我想要实现的是从第二行获取类型和头像数据并将其放入如下所示的字典中:

{'type': 'member', 'avtar': 'goldilock'}

我尝试使用内置的 csv.DictReader，我不想使用 panda 。我可以使用上述模块将 csv 文件的所有内容放入字典中，但我不知道如何选择特定行并将它们放入字典中。

以下是我到目前为止的代码，是否可以使用 DictReader 模块来实现这一点？或者我应该采用另一种方法？

import csv 
inputFile="test.csv"
with open(inputFile, "rb") as data:
    reader = csv.DictReader(data)
    for row in reader:
        print row['type']

Answer 1

如果您只想要一行，请不要使用循环。使用迭代器

reader = csv.DictReader(data)
row2 = next(reader)