从嵌套字典中选择列表中的某些值

Question

我有一个输入字典:

dict1 = {'ABC':{'ARC':0,'MRC':1,'GEW':0,'TEQ':0},'FEW':{'VEW':1,'BDE':1,'LRQ':1}}

预期输出:

new_dict={'ABC':['MRC'],'FEW':['VEW','BDE','LRQ']}

有没有办法在内部字典中以值为1的列表的形式选择内部字典的键?

Answer 1

可以遍历字典的项，如果内部值为1，则添加内部键。

dict1 = {'ABC': {'ARC': 0, 'MRC': 1, 'GEW': 0, 'TEQ': 0},
'FEW': {'VEW': 1, 'BDE': 1, 'LRQ': 1}}
new_dict = {key: [k for k, v in dict1[key].items() if v == 1] for key in dict1}
print(new_dict)

输出:

{'ABC': ['MRC'], 'FEW': ['VEW', 'BDE', 'LRQ']}