添加堆叠数组单元的元素

时间:2018-09-18 04:53:11

标签: python arrays numpy

我有一个类似numpy的堆叠数组:

[[10. 12. 15. 20. 24. 20.]
 [40. 48. 60. 20. 24. 20.]]

如何将每个元素加在一起并创建一维数组,使其变为:

[50. 60. 75. 40. 48. 40]

无论堆叠数组中有多少个列表,我都需要这样做。任何帮助将不胜感激。

2 个答案:

答案 0 :(得分:2)

使用array.sum(axis=0)

a = np.array([[10., 12., 15., 20., 24., 20.],
              [40., 48., 60. ,20., 24., 20.]])

a.sum(axis=0)

np.sum(a,axis=0)

两个都给

array([50., 60., 75., 40., 48., 40.])

答案 1 :(得分:0)

“纯Python”解决方案(即非基于numpy的解决方案)可能是:

summed_list_1D = list(map(sum, zip(*python_list_2D)))