如何找到在numpy索引值的均值

Question

假設我有一個numpy數組，如下所示，我想計算每個數組（索引0）的值（1,1,1）或索引3 （4,5,6）。有沒有可以解決這個問題的一個numpy函數？我試過numpy.mean，但它不能解決問題。

[[1,2,3,4], 
[1,2,3,5], --> = [(1+1+1)/3, (2+2+2)/3, (3+3+3)/3, (4+5+6)/3] --> [1,2,3,5] 
[1,2,3,6]] 

Answer 1

a = array([[1, 2, 3, 4], 
     [1, 2, 3, 5], 
     [1, 2, 3, 6]]) 

np.mean(a, axis=0) 

-> array([ 1., 2., 3., 5.]) 

參數axis允許您選擇跨要計算平均的方向。

Answer 2

取均值沿着第一軸 - 軸：

>>> a = np.array([[1,2,3,4], 
...    [1,2,3,5], 
...    [1,2,3,6]]) 
>>> a.mean(axis=0) 
array([ 1., 2., 3., 5.])