用于返回的是x所在的最大值的下标的值。
>>> 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])