numpy recarray.argmax()函数介绍

简介

argmax()函数是NumPy中的一个函数，其作用是返回一个给定数组的最大值的索引。当数组为多维数组时，可以指定轴以沿该轴找到最大值的索引。 当数组为结构化数组时，我们可以使用recarray类来调用该方法。

语法

numpy.recarray.argmax(axis=None, out=None)

参数

• axis：整数或元组，用于确定要跨哪个轴查找最大值的索引。
• out：用于保存结果的可选输出数组。

返回值

返回一个数组中最大值的索引。如果给定多轴，则返回轴上的最大值。

示例

假设我们有以下结构化数组：

import numpy as np

datatype = [('name', 'S10'), ('age', int), ('salary', float)]
employees = np.array([('John', 32, 5000), ('David', 25, 3000), ('Sara', 28, 8000)], dtype=datatype)

我们可以调用argmax()函数来查找具有最高薪水的员工的索引：

highest_paid_employee_index = employees['salary'].argmax()
print(highest_paid_employee_index)

# 输出：2

这里，我们首先访问结构化数组的salary字段，这将返回一个一维数组，表示每个员工的薪水。然后我们在该数组上调用argmax()函数，以获取该数组中具有最高薪水的员工的索引。在这个例子中，Sara的薪水最高，所以argmax()函数返回2。