如何对数组中的所有第n行数组（numpy）求和？

Question

假设我有阵列：

 import numpy as np
 a = np.array([[[1], [3], [5]], [[2], [8], [6]]]

如何将所有第一行汇总在一起，将所有第二行汇总在一起呢？所以，我想要的结果是这样的

 [3, 11, 11] or [[3], [11], [11]]

它似乎很简单，但我找不到不需要循环的解决方案......

Answer 1

我认为您正在寻找的是np.sum，其中您对0th轴进行求和。

import numpy as np
a = np.array([[[1], [3], [5]], [[2], [8], [6]]])
b = a.sum(0)
# b = array([[3],[11],[11]])

虽然您的初始添加已关闭1 + 2 = 3，但3 + 8 = 11和5 + 6 = 11已离开3, 11, 11

或者，正如NAN所指出的

np.sum(a, axis=(0,2))
# array([ 3, 11, 11])