将类似的嵌套dict的值合并到新的嵌套dict中

时间:2018-11-01 22:25:33

标签: python dictionary nested key-value

我必须使用的嵌套字典示例如下:

d1 = {'Breda': {'Fuery': {'anger': 1},
                       'Havoc': {'anticipation': 1, 'surprise': 1},
                       'Hawkeye': {'anticipation': 2}},
             'Fuery': {'Breda': {'anticipation': 2, 'surprise': 1},
                       'Mustang': {'fear': 4}},
             'Havoc': {'Breda': {'joy': 1},
                       'Gloria': {'joy': 1},
                       'Hawkeye': {'anticipation': 4, 'joy': 1, 'surprise': 1},
                       'Mustang': {'surprise': 1}},
             'Hawkeye': {'Hawkeye': {'joy': 2, 'trust': 1},
                         'Mustang': {'anger': 1,
                                     'joy': 6,
                                     'surprise': 4,
                                     'trust': 1}},
             'Mustang': {'Everyone': {'anger': 2},
                         'Fuery': {'anger': 3, 'surprise': 1},
                         'Hawkeye': {'anger': 1,
                                     'joy': 6,
                                     'surprise': 1,
                                     'trust': 1},
                         'Mustang': {'joy': 2, 'trust': 1},
                         'the team': {'anger': 1}}}

我正在尝试从中创建一个新的嵌套字典d2,使新键成为d1最外层字典的键,然后将每个键的值作为所有内部d1的键的值的组合。

因此,例如,如果'anticipation'显示在多个键中,则'anticipation'的值应加在一起。

所需的输出如下所示:

d2 = {'Breda': {'anger': 1, 'anticipation': 3, 'surprise': 1},
      'Fuery': {'anticipation': 2, 'surprise': 1, 'fear': 4},
      'Havoc': {'joy': 3, 'anticipation': 4, 'surprise': 2},
      'Hawkeye': {'joy': 8, 'trust': 2, 'anger': 1, 'surprise': 4}
      'Mustang': {'anger': 6, 'surprise': 2, 'joy': 8, 'trust': 2}}

到目前为止,我有这段代码不能以某种方式遍历内部dict的所有键,而只输出最后一个的值:

from pprint import pprint

      # see d1 above

d2 = dict()

for key, value in d1.items():

    for k, v in value.items():
        d2[key] = v

pprint(d2) 

当前输出:

{'Breda': {'anticipation': 2},
 'Fuery': {'fear': 4},
 'Havoc': {'surprise': 1},
 'Hawkeye': {'anger': 1, 'joy': 6, 'surprise': 4, 'trust': 1},
 'Mustang': {'anger': 1}}

如何遍历内部字典的所有键,然后比较值以将它们加在一起?

我希望我的问题清楚。

任何提示将不胜感激!

1 个答案:

答案 0 :(得分:0)

您可以在内部词典的值上使用Counter

from collections import Counter

d1 = {'Breda': {'Fuery': {'anger': 1},
                'Havoc': {'anticipation': 1, 'surprise': 1},
                'Hawkeye': {'anticipation': 2}},
      'Fuery': {'Breda': {'anticipation': 2, 'surprise': 1},
                'Mustang': {'fear': 4}},
      'Havoc': {'Breda': {'joy': 1},
                'Gloria': {'joy': 1},
                'Hawkeye': {'anticipation': 4, 'joy': 1, 'surprise': 1},
                'Mustang': {'surprise': 1}},
      'Hawkeye': {'Hawkeye': {'joy': 2, 'trust': 1},
                  'Mustang': {'anger': 1,
                              'joy': 6,
                              'surprise': 4,
                              'trust': 1}},
      'Mustang': {'Everyone': {'anger': 2},
                  'Fuery': {'anger': 3, 'surprise': 1},
                  'Hawkeye': {'anger': 1,
                              'joy': 6,
                              'surprise': 1,
                              'trust': 1},
                  'Mustang': {'joy': 2, 'trust': 1},
                  'the team': {'anger': 1}}}

result = {key: dict(sum(map(Counter, inner.values()), Counter())) for key, inner in d1.items()}

print(result)

输出

{'Havoc': {'joy': 3, 'surprise': 2, 'anticipation': 4}, 'Mustang': {'joy': 8, 'trust': 2, 'anger': 7, 'surprise': 2}, 'Hawkeye': {'joy': 8, 'surprise': 4, 'anger': 1, 'trust': 2}, 'Breda': {'surprise': 1, 'anger': 1, 'anticipation': 3}, 'Fuery': {'fear': 4, 'surprise': 1, 'anticipation': 2}}

语句sum(map(Counter, inner.values()), Counter())将每个内部字典转换为Counter,然后使用sum将它们组合在一起。之所以可以这样做,是因为Counter实现了加法,例如:

Counter({'anticipation': 2}) + Counter({'anticipation': 1, 'surprise': 1})

返回:

Counter({'anticipation': 3, 'surprise': 1})

更新

如果要每个键的总数,只需添加以下行:

total = { key : sum(inner.values()) for key, inner in result.items() }
print(total)

输出

{'Hawkeye': 15, 'Mustang': 19, 'Fuery': 7, 'Havoc': 9, 'Breda': 5}
相关问题