python - 转换字典对象

Question

这个字典结构:

data = {
    'a': { 'category': ['c', 'd'] },
    'b': { 'category': ['c', 'd'] }
}

应该变成这个字典结构:

data = {
    'c' : ['a', 'b'],
    'd' : ['a', 'b']
}

我有以下方法:

for key, value in data.items():
    if isinstance(value, dict):
        if 'category' in value:
            for cat in value['category']:
                if cat in categories:
                    categories[cat].append(key)
                else:
                    categories[cat] = [key]

我想知道是否有任何方法可以简化我的方法。我正在使用 python 3.5

Answer 1

你可以用 defaultdict(list) 来解决它，遍历类别并附加键:

>>> from collections import defaultdict
>>>
>>> d = defaultdict(list)
>>> for key, value in data.items():
...     for item in value['category']:
...         d[item].append(key)
... 
>>> dict(d)
{'c': ['a', 'b'], 'd': ['a', 'b']}