导读 在Python的数据分析领域,`numpy`是一个不可或缺的工具库,而其中的`argmin()`函数更是帮助我们快速找到数组中最小值索引的利器!✨简单来...
在Python的数据分析领域,`numpy`是一个不可或缺的工具库,而其中的`argmin()`函数更是帮助我们快速找到数组中最小值索引的利器!✨
简单来说,`argmin()`会返回数组中最小值对应的索引位置。例如,对于数组`[4, 2, 9, 1]`,`argmin()`将返回`3`(因为最小值`1`位于索引`3`)。
```python
import numpy as np
arr = np.array([4, 2, 9, 1])
print(np.argmin(arr)) 输出:3
```
此外,`argmin()`还支持处理多维数组。通过设置参数`axis`,我们可以指定按行或列的方向寻找最小值的索引。比如:
```python
matrix = np.array([[8, 5], [3, 9]])
print(np.argmin(matrix, axis=0)) 按列找最小值索引 → [1 1]
print(np.argmin(matrix, axis=1)) 按行找最小值索引 → [1 0]
```
值得一提的是,当数组中包含正负无穷时,`argmin()`也能正确识别并返回相应的位置。例如:
```python
inf_arr = np.array([np.inf, -np.inf, 0])
print(np.argmin(inf_arr)) 输出:1 (-∞是最小值)
```
掌握`argmin()`的这些特性,可以让你更高效地处理各种复杂数据场景!🌟