numpy.argsort

numpy.argsort(a, axis=-1, kind='quicksort', order=None)

返回将数组分类的索引。

使用种类关键字指定的算法沿给定轴执行间接排序。它返回与a形状相同的索引的数组，按照排序顺序沿给定轴的索引数据。

参数：

a：array_like

数组进行排序。

axis：int或None，可选

要排序的轴。默认值为-1（最后一个轴）。如果为“无”，则使用展平的数组。

kind：{'quicksort'，'mergesort'，'heapsort'}，可选

排序算法。

order：str或str的列表，可选

当a是定义了字段的数组时，此参数指定首先，第二等比较哪些字段。单个字段可以指定为字符串，并且不是所有字段都需要指定，但是未指定的字段仍将按照它们在dtype中出现的顺序使用，以断开关系。

返回：

index_array：ndarray，int

沿指定轴排序a的指数数组。如果a是一维的，则a[index_array]产生排序的a。

也可以看看

sort

描述使用的排序算法。

lexsort

使用多个键的间接稳定排序。

ndarray.sort

排序排序。

argpartition

间接部分排序。

笔记

有关不同排序算法的说明，请参见sort。

从NumPy 1.4.0 argsort起，使用包含纳米值的实数/复数数组。增强的排序顺序记录在sort中。

例子

一维数组：

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

二维数组：

>>> 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]])

使用键排序：

>>> x = np.array([(1, 0), (0, 1)], dtype=[('x', '<i4'), ('y', '<i4')])
>>> x
array([(1, 0), (0, 1)],
      dtype=[('x', '<i4'), ('y', '<i4')])

>>> np.argsort(x, order=('x','y'))
array([1, 0])

>>> np.argsort(x, order=('y','x'))
array([0, 1])