如何获取np.argsort()函数中索引最大的值
时间: 2024-04-30 16:23:45 浏览: 106
浅析python中numpy包中的argsort函数的使用
5星 · 资源好评率100%
可以使用`np.argsort()`函数来获取数组中元素从小到大的索引值,然后选择最大的索引值。具体实现如下:
```python
import numpy as np
arr = np.array([3, 2, 1, 4, 5])
sorted_idx = np.argsort(arr)
max_idx = sorted_idx[-1] # 最大索引值为排序后的最后一个元素
max_value = arr[max_idx] # 获取最大值
print("最大值的索引值为:", max_idx)
print("最大值为:", max_value)
```
输出结果为:
```
最大值的索引值为: 4
最大值为: 5
```
阅读全文