参考文章:https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.argsort.html
numpy.
argsort
(
a,
axis=-1,
kind='quicksort',
order=None
)
返回数组排序后对应的下标。kind是排序算法,axis是排序的轴。
Parameters: | a : array_like
axis : int or None, optional
kind : {‘quicksort’, ‘mergesort’, ‘heapsort’}, optional
order : str or list of str, optional
|
---|---|
Returns: | index_array : ndarray, int
|
举例:
1、一维数组
>>> x = np.array([3, 1, 2]) >>> np.argsort(x) array([1, 2, 0])
2、二维数组
>>> x = np.array([[0, 3], [2, 2]]) >>> x array([[0, 3], [2, 2]])
>>> np.argsort(x, axis=0) array([[0, 1], [1, 0]])
>>> np.argsort(x, axis=1) array([[0, 1], [0, 1]])