argmax()

源自于： numpy.random.argmax() 

傳回的是x所在的最大值的下标的值，有點抽象，下面看api中給的幾個demo:

>>> 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])
           

上面是我自己的了解，反正我是了解了，你了解了麼？