【Python】Tips：返回数组中最大值的下标np.argmax()

描述

返回数组中最大值的下标

语法

argmax(a, axis=None, out=None, *, keepdims=np._NoValue)

参数

a : array_like
Input array.
axis : int, optional
By default, the index is into the flattened array, otherwise along the specified axis.
注意：axis默认为None，将数组数据平铺，取最大值的下标返回；axis设置为0，按列寻找，返回每列最大值所在的行的下标； axis设置为1，按行寻找，返回每行最大值所在的列的下标。
out : array, optional
If provided, the result will be inserted into this array. It should be of the appropriate shape and dtype.
keepdims : bool, optional
If this is set to True, the axes which are reduced are left in the result as dimensions with size one. With this option, the result will broadcast correctly against the array.

实例

# 生成一个2*3的numpy数组
import numpy as np
a = np.arange(6).reshape(2,3) + 10

输出

a
array([[10, 11, 12],
           [13, 14, 15]])

# axis默认为None，将数组数据平铺，取最大值的下标返回
np.argmax(a)
5

# axis设置为0，按列寻找，返回每列最大值所在的行的下标
np.argmax(a, axis=0)
array([1, 1, 1])

# axis设置为1，按行寻找，返回每行最大值所在的列的下标
np.argmax(a, axis=1)
array([2, 2])