Java教程

numpy.argsort()函数

本文主要是介绍numpy.argsort()函数，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！
点击查看代码 
# argsort() 对数组升序排列，但输出的是原数组下标，不是排序好的数组

# argsort()[num]
# 当num >= 0时，argsort()[num]的值即为y[num]的值
# 当num < 0时，argsort()[num]的值为y数组反向输出的第num个数

import numpy as np

x1 = np.array([-1, 0, 1, 4, 5])
x2 = np.array([0, -1, 5, 1, 4])
y1 = x1.argsort()
y2 = x2.argsort()

print(y1)  # [0 1 2 3 4]
print(y2)  # [1 0 3 4 2]

print(y1[1])  # 1
print(x1.argsort()[1])  # 1
print(y1[-1])  # 4
print(x1.argsort()[-5])  # 0

print(y2[1])  # 0
这篇关于numpy.argsort()函数的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航