numpy.argsort(a,axis=-1)
功能:将矩阵a按照axis(即按照行还是列)排序,并返回排序后的下标
import numpy as np
x = np.array([1, 9, 7])
y=np.argsort(x)#
print(y)#[0 2 1]
numpy.argsort(a,axis=-1)
功能:将矩阵a按照axis(即按照行还是列)排序,并返回排序后的下标
import numpy as np
x = np.array([1, 9, 7])
y=np.argsort(x)#
print(y)#[0 2 1]