如何获取np.argsort()函数中索引最笑的值
时间: 2024-05-08 12:19:06 浏览: 138
浅析python中numpy包中的argsort函数的使用
5星 · 资源好评率100%
可以通过np.argsort()函数的返回值来获取索引最小的值,即返回值的第一个元素。例如:
```python
import numpy as np
a = np.array([3, 1, 4, 2, 5])
idx = np.argsort(a)
print(idx[0]) # 输出为 1,表示a中最小的值为1,其索引为1
```
需要注意的是,如果a中存在多个最小值,那么返回的是这些最小值中索引最小的一个。如果需要获取所有最小值的索引,可以使用np.where()函数。例如:
```python
import numpy as np
a = np.array([3, 1, 4, 2, 1])
min_val = np.min(a)
idx = np.where(a == min_val)[0]
print(idx) # 输出为 [1 4],表示a中所有最小的值的索引为1和4
```
阅读全文