Python学习笔记6：Numpy.argmax()

用于返回的是x所在的最大值的下标的值。

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