python topN / topK 取 最大的N个数 或 最小的N个数

import numpy as np
a = np.array([1,4,3,5,2])
b = np.argsort(a)
print(b)

print结果[0 4 2 1 3]
说明a[0]最小，a[3]最大

a[0]<a[4]<a[2]<a[1]<a[3]