2016-10-23 258 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.]]