如何获取CSV文件的密钥作为列表？

Question

我想将以下CSV文件的密钥存储为列表:

"first_name","last_name","email","mobile"
"rahul","sivadas","rahul@gmail.com",783434513
"mary","tomy","mary@gmail.com",9839383894
"vijay","govind","vijay@gmail.com",9283747393
"vikas","raj","vikas@gmail.com",239848392
"ajay","r","ajay@gmail.com",982934793

如何将其键作为列表:

['first_name','last_name','email','mobile']

我试过了:

>>> with open('test.csv', 'rb') as f:
        header = csv.header(f)
        print header



Traceback (most recent call last):
  File "<pyshell#4>", line 2, in <module>
    header = csv.header(f)
AttributeError: 'module' object has no attribute 'header'

Answer 1

该csv模块肯定没有header属性,因为错误回溯告诉您.我想你也许需要csv.DictReader的 fieldnames属性:

的字段名的参数是一个序列,其元素与输入数据的顺序中的字段相关联.如果省略fieldnames参数,则csvfile的第一行中的值将用作字段名.

正在使用:

>>> import csv
>>> with open('test.csv') as f:
    reader = csv.DictReader(f)
    print reader.fieldnames


['first_name', 'last_name', 'email', 'mobile']