当前位置:首页 >> 脚本专栏

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个数方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。