CSV行作为字典中的键/值

时间:2019-06-26 21:03:58

标签: python csv dictionary

想象一个看起来像

的CSV文件
Zipcode  State
0000     CA
0001     CA

我想创建一个词典,其中的键是邮政编码,值是状态。

我已经编写了一个函数,可以逐行读取CSV

for i, line in enumerate(reader):
    print ('line[{}] = {}'.format(i, line))

output: 
line[1] = ['0000', 'CA']
line[2] = ['0001', 'CA']

如何将每一行放入字典中? 希望输出作为回报dict[zipcode]

2 个答案:

答案 0 :(得分:0)

由于您已经对解析的行进行了迭代,因此您可以仅使用字典理解来获得所需的内容。

{line[0]: line[1] for line in reader}

答案 1 :(得分:0)

使用pandas.read_csv

的替代方法
import pandas as pd

df=pd.read_csv(file_path)
df.set_index('Zipcode').T.to_dict(orient='records')[0]

输出:

{'0000': 'CA', '0001': 'CA'}