如何找到最小/最大值由轴在numpy的阵列

Question

轴线I有一个NumPy的阵列的(3,1,2)形状：

A=np.array([[[1,4]], 
      [[2,5]], 
      [[3,2]]]). 

我想获得最小在每列中。

在这种情况下，他们是1和2。我试图用np.amin但它返回一个数组，这是不是我想要的。有没有办法做到这一点在一两行python代码没有使用循环？

Answer 1

可以指定axis作为参数传递给numpy.min功能。

In [10]: A=np.array([[[1,4]], 
       [[2,5]], 
       [[3,6]]]) 

In [11]: np.min(A) 
Out[11]: 1 

In [12]: np.min(A, axis=0) 
Out[12]: array([[1, 4]]) 

In [13]: np.min(A, axis=1) 
Out[13]: 
array([[1, 4], 
     [2, 5], 
     [3, 6]]) 

In [14]: np.min(A, axis=2) 
Out[14]: 
array([[1], 
     [2], 
     [3]])