代码片段记录19

a = np.array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])
a
array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])
ind = np.argpartition(a, -4)[-4:]
ind
array([1, 5, 8, 0])
a[ind]
array([4, 9, 6, 9])



# 下面的是排好序的

In [1]: import numpy as np

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

In [3]: arr.argsort()[-3:][::-1]
Out[3]: array([4, 3, 1])