python - 多个列表列表索引的总和

Question

假设我有一个列表列表:

[[5, 10, 30, 24, 100], [1, 9, 25, 49, 81]]
[[15, 10, 10, 16, 70], [10, 1, 25, 11, 19]]
[[34, 20, 10, 10, 30], [9, 20, 25, 30, 80]]

现在我想要第一个列表的所有索引的总和，然后是第二个列表 5+15+34=54 10+10+20=40 等等:

[54,40,50, 50,200], [20,30,75,90,180]

我试过:

for res in results:     
    print [sum(j) for j in zip(*res)] 

这里的 results 是列表的列表。 但它给出了每个列表项的总和:

[6,19,55,73,181]
[25,11,35,27,89]
[43,40,35,40,110]

Answer 1

你几乎是正确的，你需要解压 results 并压缩它。

>>> data = [[[5, 10, 30, 24, 100], [1, 9, 25, 49, 81]],
...         [[15, 10, 10, 16, 70], [10, 1, 25, 11, 19]],
...         [[34, 20, 10, 10, 30], [9, 20, 25, 30, 80]]]
>>> for res in zip(*data):
...     print [sum(j) for j in zip(*res)] 
... 
[54, 40, 50, 50, 200]
[20, 30, 75, 90, 180]

你可以简单地用列表理解来写这个

>>> [[sum(item) for item in zip(*items)] for items in zip(*data)]
[[54, 40, 50, 50, 200], [20, 30, 75, 90, 180]]