2017-11-17 244 views
-1

比方說,我有數組:如何求和數組(numpy)中所有第n行數組?

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

我如何總結所有第一排在一起,所有的第二排在一起,等等等等?所以,我的結果,我想是這樣的

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

這似乎是很簡單,但我無法找到解決方案,它不需要循環......

+0

最後一個 - '11'? 'a.sum(0)'? – Divakar

回答

0

我認爲你是什麼尋找是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 = 33 + 8 = 115 + 6 = 11留給我們3, 11, 11

此外,作爲NAN指出

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

或np.sum(a,axis =(0,2)),如果你只是想讓它像array([3,11,11]) – NaN