Numpy中的argsort()函数_x = np.array([3, 1, 2]) np.argsort(x) array([1, 2,

作者：知新_RL | 2024-05-30 22:12:39

踩

x = np.array([3, 1, 2]) np.argsort(x) array([1, 2, 0])不对呀

np.argsort

argsort函数返回的是数组值从小到大的索引值

一维数组排序：

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

>>> np.argsort(-x)  #从大到小排序
array([0, 2, 1], dtype=int64)
1
2
3
4
5
6

二维数组：

x = np.array([[0,3,8],[7,6,5],[7,1,4]])
y = np.argsort(x,axis=0)#按列排序  相对应会返回从小到大的索引（相对于所在行）
print(y)

输出：
[[0 2 2]
 [1 0 1]
 [2 1 0]]
 
y = np.argsort(x,axis=1)#按行排序  相对应会返回从小到大的索引
输出：
[[0 1 2]
 [2 1 0]
 [1 2 0]]
1
2
3
4
5
6
7
8
9
10
11
12
13
14

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/知新_RL/article/detail/648834