sargsort函数返回的是数组值从小到大的索引值
1.对于一维数组
import numpy as np
a=np.array([5,1,3])
b=np.argsort(a)
print(b)
输出结果:
[1 2 0]
2.对于二维数组[a]
argsort(a,axis=0):按列排序
argsort(a,axis=1):按行排序
import numpy as np
a=np.array([[5,3],[2,1]])
b=np.argsort(a,axis=0)
print(b)
c=np.argsort(a,axis=1)
print(c);
输出结果
[[1 1]
[0 0]]
[[1 0]
[1 0]]
3.其他
argsort(a):按升序排列
argsort(-a):按降序排列
import numpy as np
a=np.array([5,3,-2,8])
print(np.argsort(a))
print(np.argsort(-a))
输出结果:
[2 1 0 3]
[3 0 1 2]