将csv文件的第二行读入di

Question

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

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

“type”和“avtar”数据只在第二行（user1）中，对于user2和user3，这将是空的。 我要实现的是从第二行获取类型和avtar数据，并将其放入dict中，如下所示：

^{pr2}$

我试着用内置的csv.DictReader，我不想用熊猫来做这个。但是我不能把所有的内容都放到一个目录文件中

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