numpy数组排序np.argsort
官网参考：https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html
numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
功能: 将矩阵a按照axis排序，并返回排序后的下标
参数: a:输入矩阵， axis:需要排序的维度 0，按行排序，1，按列排序
返回值: 输出排序后的下标
>>> import numpy >>> help(numpy.argsort) Help on function argsort in module numpy.core.fromnumeric: argsort(a, axis=-1, kind='quicksort', order=None) Returns the indices that would sort an array. Perform an indirect sort along the given axis using the algorithm specified by the `kind` keyword. It returns an array of indices of the same shape as `a` that index data along the given axis in sorted order.
从中可以看出argsort函数返回的是数组值从小到大的索引值
>>> x = np.array([3, 1, 2]) >>> np.argsort(x) array([1, 2, 0]) >>> x = np.array([[1, 5, 7], [3, 2, 4]]) >>> np.argsort(x, axis=0) array([[0, 1, 1], [1, 0, 0]]) #沿着行向下(每列)的元素进行排序 >>> np.argsort(x, axis=1) array([[0, 1, 2], [1, 0, 2]]) #沿着列向右(每行)的元素进行排序 >>> x = np.array([3, 1, 2]) >>> np.argsort(x) #按升序排列 array([1, 2, 0]) >>> np.argsort(-x) #按降序排列 array([0, 2, 1]) >>> x[np.argsort(x)] #通过索引值排序后的数组 array([1, 2, 3]) >>> x[np.argsort(-x)] array([3, 2, 1]) # 另一种方式实现按降序排序： >>> a = x[np.argsort(x)] >>> a array([1, 2, 3]) >>> a[::-1] array([3, 2, 1])
如何根据二维数组中的某一行或者某一列排序？假设data是一个numpy.array类型的二维数组，可以利用numpy中的argsort函数进行实现，代码实例如下：
data = data[data[:,2].argsort()] #按照第3列对行排序
注意：argsort返回的只是排好序后的行索引，不会改变原数组。
按照某行进行排序，可以利用转置操作，代码如下所示：
data = data.T(data.T[:,2].argsort()).T # 按照第3行对列进行排序
也可以直接按行进行排序，代码如下：
data = data[:,data[2].argsort()]
import numpy as np data = np.array([[1,2,3,4,5], [1,2,3,6,7], [2,3,4,5,7], [3,4,5,6,7], [4,5,6,7,8]]) idex=np.lexsort([-1*data[:,2], data[:,1], data[:,0]]) #先按第一列升序，再按第二列升序，再按第三列降序.... #注意先按后边的关键词排序 sorted_data = data[idex, :]