Python中np.argmax()函数详解：获取最大值索引
发布时间：2026-10-14| 3人浏览过
本文介绍Python中numpy库的np.argmax()函数用法，用于获取数组中最大值的索引。通过示例展示一维及二维数组的使用，并详细解析axis=0（按列比较）和axis=1（按行比较）参数的区别与输出结果。
首先来看一段示例代码。
import numpy as np
a = np.array([[2, 4, 6, 1], [1, 5, 2, 9]])
print(np.argmax(a))
print(np.argmax(a, axis=0)) #竖着比较，返回行号
print(np.argmax(a, axis=1)) #横着比较，返回列号运行上述代码后，控制台输出结果如下。
7 [0 1 0 1] [2 3]
Python np.argmax()函数的具体用法
该函数的核心功能是返回数组中最大值所对应的索引位置。
具体案例演示如下。
以上内容为个人经验分享，旨在提供参考，欢迎大家支持。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
