以一维数组为例,np.argsort()获得的是原数组升序或降序的索引排序。
获得直观的升序或降序的数组还需要对原数组进行索引,索引来源于np.argsort()的结果。
例如:
import numpy as np
a = np.array([3,1,2,1,3,5])
c = np.array([0,2,3,4,5,2])
b = np.argsort(a)
a = a[b]
c = c[b]
print(b, a, c)
output:
[1 3 2 0 4 5] [1 1 2 3 3 5] [2 4 3 0 5 2]