首页 动态 > 数码知识问答 > 正文

😊 numpy中的argmin函数基本用法 | argmin函数与无穷

导读 在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()`的这些特性,可以让你更高效地处理各种复杂数据场景!🌟

郑重声明:本文版权归原作者所有,转载文章仅为传播更多信息之目的,如作者信息标记有误,请第一时间联系我们修改或删除,多谢。