在 python 中，如何反转二维字典？

Question

我有一本如下形式的字典

dict = {
   "a" : {"a1" : 1},
   "b" : {"a2" : 1, "a3" : 2},
   "c" : {"a2" : 3, "a4" : 3}
}

我需要反向索引字典，格式如下：

inverseDict = {
    "a1" : {"a" : 1},
    "a2" : {"b" : 1, "c" : 3},
    "a3" : {"b" : 2},
    "a4" : {"c" : 3}
}

基本上

inverseDict = {dict.value.key : { dict.key : dict.value.value}}

所以本质上，我需要将值的键作为键，将键作为值的键，同时加入重复新键的结果等。

我试过了

ks = dict.keys()
vals = dict.values()

ks2 = vals.keys()
vals2 = vals.values()

如果这有任何意义

但是我遇到了一个错误

'dict_values' object has no attribute 'keys'

据我了解，这是因为 dict.values() .keys() .items() 返回“视图”而不是实际元素本身，但我不知道如何解决这个问题。

还有一个更有效的解决方案我应该考虑，因为我的实际字典非常大（约 10k 个键），并且生成的逆字典也很大（>3k 个键）

Answer 1

使用collections.defaultdict(dict) 和双循环相当简单：

d = {
    "a" : {"a1" : 1},
    "b" : {"a2" : 1, "a3" : 2},
    "c" : {"a2" : 3, "a4" : 3},
}

import collections

inverted = collections.defaultdict(dict)

for key,subd in d.items():
    for k,v in subd.items():  # no inspiration for key/value names...
        inverted[k][key] = v

inverted是

{'a1': {'a': 1},
 'a2': {'b': 1, 'c': 3},
 'a3': {'b': 2},
 'a4': {'c': 3}}

使用defaultdict 避免测试条目是否已经存在，如果不存在则创建字典值。因此，只需按预期顺序无脑添加键/值即可。

请注意，那些需要将项目处理成多个对象的问题很难使用推导式解决。

Answer 2

另一种没有标准库的解决方案......但让-弗朗索瓦·法布尔的回答更简洁，可能更容易模块化。如有疑问，请使用标准库。

OriginalDict = ... (Original dict items)
InvertedDict = {}
for k, v in OriginalDict.items():
    for k_, v_ in v.items():
        if InvertedDict.get(k_) is None:
            InvertedDict[k_] = {}
        InvertedDict[k_][k] = v_

Answer 3

你可以使用setdefault:

d = {
    'a': {'a1': 1},
    'b': {'a2': 1, 'a3': 2},
    'c': {'a2': 3, 'a4': 3}
}

result = {}
for ok, vs in d.items():
    for ik, v in vs.items():
        result.setdefault(ik, {})[ok] = v

print(result)

输出

{'a4': {'c': 3}, 'a1': {'a': 1}, 'a2': {'c': 3, 'b': 1}, 'a3': {'b': 2}}

函数setdefault与defaultdict的用法类似。