2016-10-23 219 views
1

尋找最小值最大值我有以下numpy的ndarray,形狀爲(4,1,2):在3D numpy的陣列

myarray = np.array([[[0.,4.]], 
        [[1.,5.]], 
        [[2.,6.]], 
        [[3.,7.]]]) 

如何找到最大值,最小值每一列的? 在這種情況下,第一列的min,max爲0,3;第二列的最小,最大值是4,7。

我不能完全弄清楚在這些情況下np.amin和np.amax的正確語法。

謝謝。

+2

嘗試使用'那些axis' PARAM。再次閱讀文檔。 – Divakar

回答

1
import numpy as np 

myarray = np.array([[[0., 4.]], 
        [[1., 5.]], 
        [[2., 6.]], 
        [[3., 7.]]]) 
maxes = np.max(myarray,axis=0) 
mins = np.min(myarray,axis=0) 
print 'maxes are :' ,maxes ,'\nmins are : ', mins 

這給:

maxes are : [[ 3. 7.]] 
mins are : [[ 0. 4.]]