python中将array数组进行排序并获取排序后的索引:argsort函数

argsort()函数是将x中的元素从小到大排列,提取其对应的index(索引)

  • 当num>=0时,np.argsort()[num]就可以理解为y[num]
  • 当num<0时,np.argsort()[num]就是把数组y的元素反向输出
import numpy as np
x = np.array([2,1,4,5,7,3,6]
y = x.argsort()
print (y[1], y[-1])

猜你喜欢

转载自blog.csdn.net/longwei92/article/details/83098289