版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
功能:
将矩阵a按照axis排序,并返回排序后的下标
参数:
a:输入矩阵, axis:需要排序的维度
返回值:
输出排序后的下标
>>> x = np.array([3, 1, 2])
>>> np.argsort(x)
array([1, 2, 0])
- 1
- 2
- 3
>>> 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]]) #沿着列向右(每行)的元素进行排序
- 1
- 2
- 3
- 4
- 5
- 6
- 7
以上内容来自于官方API文档