其实这个功能自己写很容易,不过有有系统自带的库,这不是很爽。
使用numpy中argsort函数。
## 获取numn数组中的N个最大值的索引?
import numpy as np
arr = np.array([1, 3, 2, 4, 5])
arr.argsort()[-3:][::-1]
返回结果如下:
Out[3]: array([4, 3, 1])
其实这个功能自己写很容易,不过有有系统自带的库,这不是很爽。
使用numpy中argsort函数。
## 获取numn数组中的N个最大值的索引?
import numpy as np
arr = np.array([1, 3, 2, 4, 5])
arr.argsort()[-3:][::-1]
返回结果如下:
Out[3]: array([4, 3, 1])