将csv文件的第二行读入dict

Question

我有一个看起来像这样的csv文件：

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

'type'和'avtar'数据仅在第二行（user1）中，对于user2和user3，这将是空的。 我想要实现的是从第二行获取类型和avtar数据并将其放入一个看起来像这样的字典：

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

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

以下是我到目前为止的代码，是否可以使用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)