添加字典值的元素

Question

我有一个字典，看起来像下面的

output_dict = {"weight_1": [1, 2, 3], "weight_2": [4, 5, 6]}

我想要一个看起来像这样的结果列表

results_list = [5, 7, 9]

这也需要适用于n个权重，例如，如果我们有

output_dict = {"weight_1": [1, 2, 3], "weight_2": [4, 5, 6], "weight_3": [7, 8, 9]}

输出为

results_list = [12, 15, 18]

Answer 1

zip字典值，然后sum他们：

[sum(vals) for vals in zip(*output_dict.values())]

或者使用map：

[*map(sum, zip(*output_dict.values()))]