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

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])