numpy().argsort()函数

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]