np.argsort函数

最新推荐文章于 2024-06-21 17:30:00 发布

武科大许志伟

最新推荐文章于 2024-06-21 17:30:00 发布

阅读量2k

点赞数

分类专栏： python 文章标签： np.argsort numpy 排序后返回索引

本文链接：https://blog.csdn.net/u013555719/article/details/83864122

版权

python 专栏收录该内容

70 篇文章 2 订阅

订阅专栏

np.argsort函数

觉得有用的话,欢迎一起讨论相互学习~

numpy.argsort(a, axis=-1, kind=’quicksort’, order=None)
功能: 将矩阵a按照axis排序，并返回排序后的下标
参数: a:输入矩阵， axis:需要排序的维度
返回值: 输出排序后的下标

import numpy as np
x1 = np.array([3, 1, 2])
print(np.argsort(x1))
# [1 2 0]

# axis=0
#沿着行向下(每列)的元素进行排序
x2 = np.array([[1, 5, 7], [3, 2, 4]])
print("axis=0排序\n",np.argsort(x2, axis=0))
# axis=0排序
#  [[0 1 1]
#  [1 0 0]]

# axis=1
#沿着列向右(每行)的元素进行排序
print("axis=1排序\n",np.argsort(x2, axis=1))
# axis=1排序
#  [[0 1 2]
#  [1 0 2]]