說明

numpy.argsort(a, axis=-1, kind='quicksort', order=None) 

• 功能: 將矩陣a按照axis排序，並返回排序後的下標
• 引數: a:輸入矩陣， axis:需要排序的維度
• 返回值: 輸出排序後的下標

例項

>>> x = np.array([3, 1, 2])
>>> np.argsort(x)
array([1, 2, 0])

沿著行向下(每列)的元素進行排序

>>> x = np.array([[1, 5, 7], [3,
 
 2, 4]])
>>> np.argsort(x, axis=0)
array([[0, 1, 1],
       [1, 0, 0]])  #沿著行向下(每列)的元素進行排序

沿著列向右(每行)的元素進行排序

>>> np.argsort(x, axis=1)
array([[0, 1, 2],
       [1, 0, 2]])  #沿著列向右(每行)的元素進行排序