numpy.argmax(a, axis=None, out=None)
返回沿轴axis最大值的索引
Parameters:
a : array_like -数组
axis : int, 可选 -默认情况下，索引的是平铺的数组，否则沿指定的轴。
out : array, 可选 -如果提供，结果以合适的形状和类型被插入到此数组中。
Returns:
index_array : ndarray of ints -索引数组。它具有与a.shape相同的形状，其中axis被移除。
例子：
1 >>> a = np.arange(6).reshape(2,3) 2 >>> a 3 array([[0, 1, 2], 4 [3, 4, 5]]) 5 >>> np.argmax(a) 6 5 7 >>> np.argmax(a, axis=0)#0代表列 8 array([1, 1, 1]) 9 >>> np.argmax(a, axis=1)#1代表行 10 array([2, 2]) 11 >>> 12 >>> b = np.arange(6) 13 >>> b[1] = 5 14 >>> b 15 array([0, 5, 2, 3, 4, 5]) 16 >>> np.argmax(b) # 只返回第一次出现的最大值的索引 17 1