numpy.argmax()

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被移除。

例子：

 >>> a = np.arange(6).reshape(2,3)
 >>> a
 array([[0, 1, 2],
        [3, 4, 5]])
 >>> np.argmax(a)
 5
 >>> np.argmax(a, axis=0)#0代表列
 array([1, 1, 1])
 >>> np.argmax(a, axis=1)#1代表行
 array([2, 2])
 >>>
 >>> b = np.arange(6)
 >>> b[1] = 5
 >>> b
 array([0, 5, 2, 3, 4, 5])
 >>> np.argmax(b) # 只返回第一次出现的最大值的索引
 1