將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文件的所有內容放入一個字典中，但我不知道如何選擇一個特定的行並將它們放入一個字典中。

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