NumPy argmin()函数详解
更新时间：2025年11月24日 11:11:36 作者：Geoking.
在数据分析、图像处理或机器学习任务中,常常需要找到数组或矩阵中最小值的位置,NumPy提供的argmin函数就能帮我们完成,本文将系统介绍 argmin() 的作用、用法、常见坑点,感兴趣的可以了解一下
在数据分析、图像处理或机器学习任务中，常常需要找到数组或矩阵中最小值的位置。NumPy 提供的 argmin() 函数就能帮我们完成。本文将系统介绍 argmin() 的作用、用法、常见坑点，以及与 argmax() 的关系。
一、argmin 是什么？
argmin 是 “argument of the minimum” 的缩写，
意思是 最小值对应的索引。
它返回的不是最小值，而是最小值的索引位置。
👉官方文档介绍地址numpy.argmin
二、基本语法
numpy.argmin(a, axis=None)
| 参数 | 说明 |
|---|---|
| a | 输入数组或矩阵 |
| axis | 指定沿哪个轴寻找最小值的索引，默认为 None（即对整个数组展平处理） |
返回值：
👉 一个整数（如果 axis=None）或包含索引的数组。
三、基础示例
2、一维数组
import numpy as np
arr = np.array([10, 25, 8, 36, 5])
index = np.argmin(arr)
print("最小值索引：", index)
print("最小值：", arr[index])
输出：
最小值索引： 4
最小值： 5
解释：最小值为 5，位于索引 4 位置。
2、二维数组
arr = np.array([
[3, 7, 1],
[9, 5, 2]
])
（1）不指定 axis
np.argmin(arr)
输出：
2
因为数组展平成 [3,7,1,9,5,2]，最小值 1 的索引为 2。
（2）按行查找（axis=1）
np.argmin(arr, axis=1)
输出：
[2, 2]
解释：
- 第一行最小值 1 在索引 2
- 第二行最小值 2 也在索引 2
（3）按列查找（axis=0）
np.argmin(arr, axis=0)
输出：
[0, 1, 0]
解释：
- 第 1 列最小值在第 1 行（索引 0）
- 第 2 列最小值在第 2 行（索引 1）
- 第 3 列最小值在第 1 行（索引 0）
四、argmin 与 min 的区别
| 函数 | 返回内容 | 示例 |
|---|---|---|
| np.min() | 最小值 | np.min([2,8,3]) → 2 |
| np.argmin() | 最小值的索引 | np.argmin([2,8,3]) → 0 |
两者可以搭配使用：
arr = np.array([2, 8, 3])
print("最小值：", np.min(arr))
print("最小值索引：", np.argmin(arr))
输出：
最小值： 2
最小值索引： 0
五、在机器学习中的应用
argmin() 在机器学习中也很实用，尤其在模型评估或聚类分析中。
🎯 示例：选择误差最小的模型参数
import numpy as np
errors = np.array([0.32, 0.28, 0.45, 0.25, 0.31])
best_index = np.argmin(errors)
print("误差最小的模型索引：", best_index)
print("最小误差：", errors[best_index])
输出：
误差最小的模型索引： 3
最小误差： 0.25
解释：
模型 3 的误差最小，因此最优。
六、argmin 与 argmax 的对比
| 功能 | 函数 | 返回内容 | 应用场景 |
|---|---|---|---|
| 找最大值索引 | np.argmax() | 最大值的位置 | 分类预测、最大概率标签 |
| 找最小值索引 | np.argmin() | 最小值的位置 | 最小误差、最优损失点 |
在模型训练或参数调优中，我们通常会用：
argmin(losses)：找到损失最小的 epoch；argmax(accuracy)：找到准确率最高的模型。
七、常见坑点总结
| 问题 | 错误示例 | 正确做法 |
|---|---|---|
| 忘记指定 axis 导致结果不符合预期 | np.argmin(matrix) | 指定 axis=1 或 axis=0 |
| 想取最小值却用 argmin | np.argmin(x) | 改为 np.min(x) |
| 想取二维位置但忘记转换 | 直接打印索引 | 使用 np.unravel_index() |
到此这篇关于NumPy argmin()函数详解的文章就介绍到这了,更多相关NumPy argmin()内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
您可能感兴趣的文章:
相关文章
今天小编大家分享一篇PYQT5实现控制台显示功能的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-06-06
这篇文章主要介绍了Python 实现数据结构-循环队列的操作方法,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了Python星号*与**用法,结合实例形式较为详细的分析了Python中的星号*与**在函数参数及数值运算中的相关使用技巧,需要的朋友可以参考下2018-02-02
这篇文章主要介绍了python类方法和静态方法,实例分析了Python中返回一个返回值与多个返回值的方法,需要的朋友可以参考下2021-10-10
python-pinyin是一个用于汉字转拼音的Python库,支持多音字、多种拼音风格和自定义词典,本文就来介绍一下Python中pinyin库实现汉字转换为拼音,感兴趣的可以了解一下2025-01-01
这篇文章主要介绍了Django websocket原理及功能实现代码,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
Python自动化办公Excel模块openpyxl原理及用法解析
这篇文章主要介绍了Python自动化办公Excel模块openpyxl原理及用法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
这篇文章主要介绍了Python 内置函数sorted()的用法,文章内容介绍详细具有一的参考价值，需要的小伙伴可以参考一下，希望对你的学习有所帮助2022-03-03
这篇文章主要介绍了PyTorch的Optimizer训练工具的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
在本篇文章里小编给大家整理的是一篇关于Python3爬虫ChromeDriver的安装实例内容，有兴趣的朋友们可以学习参考下。2021-02-02
最新评论