import numpy as np
a = np.array([1,4,3,5,2])
b = np.argsort(a)
print(b)
结果:[0 4 2 1 3]
如果既要得到位置,也要得到值,可以如下:
sortLoc = np.argsort(predicted_vector)
sortVal = np.sort(predicted_vector)
最后
以上就是甜美糖豆最近收集整理的关于python topN 取 最大的N个数 或 最小的N个数的全部内容,更多相关python内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复