page contents

如何在NumPy数组中获得N个最大值的索引?

轩辕小不懂 发布于 2021-07-09 13:40
阅读 125
收藏 0
分类:Python开发
1245
Nen
Nen
- 程序员

我们可以使用以下代码获取NumPy数组中N个最大值的索引:

import numpy as np 

arr = np.array([1,3,2,4,5])

print(arr.argsort()[ -  3:] [::  -  1])

请先 登录 后评论