1. 程式人生 > >np.argsort函式

np.argsort函式

np.argsort函式

覺得有用的話,歡迎一起討論相互學習~Follow Me

  • numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
  • 功能: 將矩陣a按照axis排序,並返回排序後的下標
  • 引數: a:輸入矩陣, axis:需要排序的維度
  • 返回值: 輸出排序後的下標
import numpy as np
x1 = np.array([3, 1, 2])
print(np.argsort(x1))
# [1 2 0]

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