python topN 取最大的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]

以上这篇python topn 取最大的n个数或最小的n个数方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。