python中.argsort使用举例

argsort是Numpy的函数，可以用来返回数组排序后的索引值。

举个例子，如果有一个一维数组a=[10,4,6,8,1]，我们想要将它排序后的索引值返回，就可以使用argsort函数。

import numpy as np

a = np.array([10, 4, 6, 8, 1])
idx = np.argsort(a)
print(idx)

运行结果为：

[4 1 2 3 0]

其中，idx返回的是a数组排序后的索引值，也就是说，a[4]是最小值，a[0]是最大值。

我们还可以使用argsort函数按照多个条件排序，如下所示：

import numpy as np

a = np.array([10, 4, 6, 8, 1])
b = np.array([4, 3, 1, 2, 5])
idx = np.lexsort((a, b))
print(idx)

运行结果为：

[2 3 1 0 4]

其中，lexsort函数按照最后一个传入的数组b排序，如果有相同的值，再按照前面传入的a数组排序。idx返回的是排序后的索引值。