将csv数据存储在字典中

Question

我正在使用 python 3。我在 csv 文件中有示例数据，我想将其转换为字典：

a.csv 中的数据：

a.fruit,a.colour,a.quantity 
apple,red,2 
grapes,purple,4
pear,green,5

我有以下代码：

sample = 'samples.csv'
file_name = open(sample,'r')

reader = csv.DictReader(file_name)
for row in reader:
    print(row)  

当我运行代码时，我得到以下输出：

['a.fruit':'apple','a.colour':'red','a.quantity':'2']
['a.fruit':'grapes','a.colour':'purple','a.quantity':'4']
['a.fruit':'pear','a.colour':'green','a.quantity':'5']

我有没有得到以下字典：

[{'a.fruit':['apple','grapes','pear'],'a.colour':['red','purple','green'],'a.quantity':['2','4','5'}]

所以存储在 (str: list of str) for (key:value) 中的 dict 表示？

另外，是否可以输出以下内容： [a : {'a.fruit':['apple','grapes','pear'],'a.colour':['red','purple','green'],'a.quantity': ['2','4','5'}]

Answer 1

你可以像这样使用list comprehension和dict comprehension

dicts = [row for row in reader]
result = {k:[d[k] for d in dicts] for k in dicts[0].keys()}
print(result)

输出

{'a.fruit': ['apple', 'grapes', 'pear'], 'a.colour': ['red', 'purple', 'green'], 'a.quantity ': ['2 ', '4', '5']}