Python 如何在numpy数组中逐轴查找最小值/最大值
我有一个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]:
(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]])