numpy.argmax/argmin/max/min

查找数组最大值最小值,以及他们的索引。

使用对象方法

In [1]: import numpy as np
   ...: a=np.array([1,2,3,3,2,1])

In [2]: a.max() 
Out[2]: 3

In [3]: a.argmax() #返回第一个最大值的索引
Out[3]: 2

In [4]: a.min()
Out[4]: 1

In [5]: a.argmin() #返回第一个最小值的索引
Out[5]: 0

使用类方法

In [6]: np.max(a)
Out[6]: 3

In [7]: np.argmax(a)
Out[7]: 2

In [8]: np.min(a)
Out[8]: 1

In [9]: np.argmin(a)
Out[9]: 0

你可能感兴趣的:(numpy)