np.argsort()函数

在jupyter notebook 中同时输入多行

from IPython.core.interactiveshell import InteractiveShell
InteractiveShell.ast_node_interactivity = 'all'

np.argsort(a, axis=-1, kind=‘quicksort’, order=None)

返回将对数组进行排序的索引。

参数

• a : 进行排序的数组
• axis : 可选参数，按axis轴排序
• kind : {‘quicksort’, ‘mergesort’, ‘heapsort’, ‘stable’}, 可选参数
• order : str or list of str, optional

返回值

返回数组排序后的索引

参看

• sort：描述使用的排序算法。
• lexsort：具有多个键的间接稳定排序。
• ndarray.sort：就地排序。
• argpartition：间接局部排序。

实例一：

• 一维数据

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