Python基本函数:np.argmax（

Python基本函数:np.argmax（）

• 一、函数说明
• 二、函数用法

 

格式：np.argmax(a)
注意：返回的是a中元素最大值所对应的索引值

一、函数说明

argmax(a, axis=None, out=None)
    Returns the indices of the maximum values along an axis.           #返回向量的最大值的索引

    Parameters
    ----------
    a : array_like
        Input array.
    axis : int, optional
        By default, the index is into the flattened array, otherwise
        along the specified axis.
    out : array, optional
        If provided, the result will be inserted into this array. It should
        be of the appropriate shape and dtype.

  返回元素最大值所对应的索引值！若还是没看明白，下面则是例子来辅助理解。
 

二、函数用法

1、一维数组

In :       	a = np.array([3,1,2,1,3,5])
Out: 		[3,1,2,1,3,5]

In : 		b = np.argmax(a)                        # 元素最大值的索引值      
Out: 		5    

2、二维数组

In :       	a = np.array([[1, 3, 5, 7],[5, 7, 2, 2],[4, 6, 8, 1]])
Out: 		[[1, 3, 5, 7],
             [5, 7, 2, 2],
             [4, 6, 8, 1]]

In : 		b = np.argmax(a, axis=0)						# 对数组按列方向搜索最大值                           
Out: 		[1 1 2 0]       

In : 		b = np.argmax(a, axis=1)						# 对数组按行方向搜索最大值                           
Out: 		[3 1 2]