原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html
校对:(虚位以待)
numpy.
argsort
(a, axis=-1, kind='quicksort', order=None)[source]返回将数组分类的索引。
使用种类关键字指定的算法沿给定轴执行间接排序。它返回与a形状相同的索引的数组,按照排序顺序沿给定轴的索引数据。
参数: | a:array_like
axis:int或None,可选
kind:{'quicksort','mergesort','heapsort'},可选
order:str或str的列表,可选
|
---|---|
返回: | index_array:ndarray,int
|
也可以看看
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])