16
2020
03

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

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


importnumpy as np


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


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


4 3 1

« 上一篇 下一篇 »

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。