源自于: numpy.random.argmax()
返回的是x所在的最大值的下标的值,有点抽象,下面看api中给的几个demo:
>>> a = np.arange(6).reshape(2,3)
>>> a
array([[0, 1, 2],
[3, 4, 5]])
#默认是返回的所有的元素的最大的一个下标
>>> np.argmax(a)
5
#axis = 0 ,表示以行为基准,比较在相同列上的元素的大小返回下标,即[0,3],[1,4],[2,5]
>>> np.argmax(a, axis=0)
array([1, 1, 1])
#axis = 0, 表示以列为基准,比较在相同行上的元素的大小返回下标,即[0,1,2], [3,4,5]
>>> np.argmax(a, axis=1)
array([2, 2])
上面是我自己的理解,反正我是理解了,你理解了么?