python数据分析numpy基础之argmax求数组最大值索引

1 python数据分析numpy基础之argmax求数组最大值索引

python的numpy库的argmax()函数，用于获取沿指定轴的最大值的索引。

用法

numpy.argmax(a, axis=None, out=None, *, keepdims=<no value>)

描述

argmax()返回沿指定轴的最大值的索引。

入参axis表示指定轴，默认为None表示整个数组的最大值的索引而不是某个轴的最大值索引，即转为一维数组后的最大值的索引。axis只能为整数，不支持整数元组。

入参keepdims表示是否保留指定轴的尺寸为1，默认为False，不保留。

1.1 入参a

numpy.argmax()的入参a，为必选入参，可以为数组、列表、元组。

>>> import numpy as np
# 入参a为列表
>>> np.argmax([-5,-10,0,1,5,39,2])
5
# 入参a为元组
>>> np.argmax((-5,-10,0,1,5,39,2))
5
# 入参a为数组
>>> np.argmax(np.array((-5,-10,0,1,5,39,2)))
5

1.2 入参axis为整数

numpy.argmax()的入参axis为可选入参，只能为整数（不支持元组），范围[0, ndim)，默认为None，表示求整个数组的最大值的索引，若为多维数组则转换为一维数组来获取最大值的索引。

若axis=n为整数，则对沿指定轴n的元素求最大值的索引，并且只返回指定轴的索引，其他轴索引不返回。

若axis=负数，则-1对应最后一个轴，-2倒数第2个轴，依此类推。

最大值多次出现时，argmax()返回第一次出现的索引。

np.ndindex(ar3.shape) 获取指定形状数组的索引迭代器对象，遍历迭代器索引获取数组元素。

>>> import numpy as np
>>> ar3=np.array([[[12, 13, 8, 15 ],
        [ 19,  25,  16,  5],
        [ 18,  29, 10, 31]],

       [[9,  11,  21,  3],
        [16, 17, 31, 19],
        [20, 3, 22, 23]]])
# axis默认为None，返回整个数组的最大值的索引
>>> np.argmax(ar3)
11
# ar3最大值为31
>>> np.max(ar3)
31
# axis=None时，ar3相当于一维数组
>>> ar3.reshape