Python 如何在numpy数组中逐轴查找最小值/最大值

Python 如何在numpy数组中逐轴查找最小值/最大值,python,arrays,numpy,Python,Arrays,Numpy,我有一个NumPy数组,形状为(3,1,2): 我想得到每列的最小值 在本例中,它们是1和2。我试图使用np.amin,但它返回一个数组,这不是我想要的。有没有一种方法可以在一行或两行python代码中实现这一点,而不使用循环?您可以将axis指定为numpy.min函数的参数 In [10]: A=np.array([[[1,4]], [[2,5]], [[3,6]]]) In [11]: np.min(A) Out[11]:

我有一个NumPy数组,形状为
(3,1,2)

我想得到每列的最小值


在本例中,它们是1和2。我试图使用
np.amin
,但它返回一个数组,这不是我想要的。有没有一种方法可以在一行或两行python代码中实现这一点,而不使用循环?

您可以将
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]])

可能是@mengmengxyz的副本,我不确定我是否理解您的问题
np.min(np.array([[1,4]],[[2,5]],[[3,2]]),axis=0
生成:
array([[1,2]])
,这不是你要找的吗?这很有效。。对不起,把你弄糊涂了。我在测试时把代码弄乱了。谢谢
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]])