python - 在列表组件列表中明智地添加或减去元素

Question

如何在列表组件中明智地添加列表元素？

p=[[1,2,3],[1,0,-1]]

我尝试了以下方法:

list(map(sum,zip(p[0],p[1])))

会得到我需要的 [2,2,2]。但是如何将它扩展为可变数量的列表呢？例如，p=[[1,2,3],[1,0,-1],[1,1,1]] 应该产生 [3,3,3] 。

我想到的解决方案如下:

import pandas as pd
p=[[1,2,3],[1,0,-1],[1,1,1]]
list(pd.DataFrame(p).sum())

是否有更“Pythonic”的方式来解决这个问题？

Answer 1

使用* 解包列表:

a = list(map(sum,zip(*p)))
print (a)
[3, 3, 3]

在 numpy 中的解决方案与在 pandas 中类似:

a = np.array(p).sum(axis=0).tolist()
print(a)
[3, 3, 3]