Numpy argmax()函数用于返回给定数组中的最大值所在的索引位置。

它的语法格式如下：

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

参数说明：

arr：传入的待计算数组，必须为一维或多维数组。

axis：可选参数，用于指定在哪个维度上进行计算，其取值范围为0到N-1（N为数组的维度数）。

out：可选参数，用于指定输出结果的数组。

如果不指定axis，则按展开后数组索引的最大值计算。例如，对于一个2×3的二维数组，展开后最大值所在的索引为3，则返回3。

下面给出两个使用实例来说明argmax()函数的用法：

输出结果为：

输出结果为：

在这个例子中，我们生成了一个3×4×5的三维数组，并使用axis=2来指定在最后一个维度中计算。输出结果中，每个矩阵在最后一个维度上的最大值所在的索引被返回了。

本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：详解Numpy argmax()（返回数组元素的最大值的索引）函数的作用与使用方法 - Python技术站