1. 程式人生 > >numpy中實用但不常見的方法(3)np.argsort

numpy中實用但不常見的方法(3)np.argsort

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]]) #沿著列向右(每行)的元素進行排序