Question

我是刚开始学习Python的新手。

我有一个CSV文件，如下所示：

theme,type,name
food,meat,bacon
food,meat,lamb
food,meat,beef
food,fruit,apple
food,fruit,pear
food,fruit,banana
food,fruit,kiwi
animal,reptile,lizard
animal,reptile,snake
animal,mammal,mouse
animal,mammal,cat

我正在尝试将其转换为Python字典，最佳输出应如下所示：

{food:{meat:[bacon,lamb],fruit:[apple, pear, banana, kiwi]},animal:{reptile:{[lizard,snake],mammal:[mouse,cat]}

我的代码：

import csv

testDict={}

with open('file.CSV','r') as f:
    tempDict = csv.DictReader(f)
    for getPKvalue in tempDict:
        testDict.setdefault(getPKvalue['theme'],[]).append(getPKvalue['type'])

print(testDict)

我尝试了上面的代码及其变体，但是没有成功。我知道我可能需要进行两个循环左右才能使其工作，但我不确定如何实现。

提前谢谢！抱歉，如果我在说明中使用了错误的术语。

Poppel

Answer 1

您非常接近。您需要列表字典的字典。

尝试：

import csv

testDict = {}
with open(filename) as f:
    tempDict = csv.DictReader(f)
    for row in tempDict:
        testDict.setdefault(row['theme'],{}).setdefault(row['type'], []).append(row['name'])
print(testDict)

输出：

{'animal': {'mammal': ['mouse', 'cat'], 'reptile': ['lizard', 'snake']},
 'food': {'fruit': ['apple', 'pear', 'banana', 'kiwi'],
          'meat': ['bacon', 'lamb', 'beef']}}

用两个键将CSV转换为字典

1 个答案: