how to use argsort in python code example

Example 1: .argsort() python

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

>>> ind = np.argsort(x, axis=1)  # sorts along last axis (across)
>>> ind
array([[0, 1],
       [0, 1]])
>>> np.take_along_axis(x, ind, axis=1)  # same as np.sort(x, axis=1)
array([[0, 3],
       [2, 2]])

Example 2: python argsort

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