numpy.argmax(a, axis=None, out=None)
返回沿軸axis最大值的索引敲街。
Parameters:
- a : array_like 數(shù)組
- axis : int, 可選
默認(rèn)情況下嫩海，索引的是平鋪的數(shù)組搂抒，否則沿指定的軸鬓长。- out : array, 可選
如果提供，結(jié)果以合適的形狀和類(lèi)型被插入到此數(shù)組中门躯。- Returns: index_array : ndarray of ints
索引數(shù)組钻蹬。它具有與a.shape相同的形狀，其中axis被移除贮缕。
示例:
>>> 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) # 只返回第一次出現(xiàn)的最大值的索引
1