python - 反转非唯一字典

Question

我有一本字典，我需要根据非唯一值进行反转和分组，但根据现有的顶级 SO 答案失败。

>>> graph = { 'a': ['car','red'], 'b': ['car','blue'] }
>>> inv_map = {}
>>> for k,v in graph.items():
        inv_map[v] = inv_map.get(v,[])
        inv_map[v].append(k)

TypeError: unhashable type: 'list'

>>> isinstance(graph, dict)
True

建议？

Answer 1

由于值是列表，您需要迭代这些列表以累积键:

from collections import defaultdict
inv_map = defaultdict(list)

for k, vs in graph.items():
    for v in vs:
        inv_map[v].append(k)

inv_map.default_factory = None  # quack like a normal dict