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]