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

例子：

 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