1. 程式人生 > >python中將array陣列進行排序並獲取排序後的索引:argsort函式

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])