python中argsort()函数的使用

argsort()函数是将x中的元素从小到大排列，提取其对应的index(索引)，然后输出到y。

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

直接上代码

import numpy as np
#argsort()函数是将x中的元素从小到大排列，提取其对应的index(索引)，然后输出到y
#一维数组
x=np.array([3,1,2])
print(np.argsort(x))#按升序排列
print(np.argsort(-x))#按降序排列
#通过索引值排序后的数组
print(x[np.argsort(x)])
print(x[np.argsort(-x)])
y=x[np.argsort(x)]
#按降序排列
print(y[::-1])




#二维数组
x2=np.array([[1,2],[3,4]])
print(np.argsort(x2))#输出[[0 1][0 1]]
#按列排序
print(np.argsort(x2,axis=0))
#按行排序
print(np.argsort(x2,axis=1))