来自多个列表(超过两个)的 Python 求和值

Question

寻找一种 pythonic 方法来对多个列表中的值求和:我有以下列表:

a = [0,5,2]
b = [2,1,1]
c = [1,1,1]
d = [5,3,4]
my_list = [a,b,c,d]

我正在寻找输出:

[8,10,8]

我用过:

print ([sum(x) for x in zip(*my_list )])

但 zip 仅在我的 my_list 中有 2 个元素时才有效。有什么想法吗？

Answer 1

zip适用于任意数量的迭代器:

>>> list(map(sum, zip(*my_list)))
[8, 10, 8]

当然，这大致等同于您的理解也有效:

>>> [sum(x) for x in zip(*my_list)]
[8, 10, 8]