python中argsort()函数的用法

格式: y = argsort(x)
x,y都是array数组

作用: 将x中的元素从小到大排列,提取其对应的index(索引),然后输出到y

这个函数是在实现k近邻算法时,需要统计距离最小的k个点,所以需要使用argsort函数将x矩阵(存着距离的数组)排序之后将索引存到y中,这样在将之后选择他是对应哪个目标变量有着一定的作用

举例:
x数组是[1, 4, 3, -1, 6, 9]
将x重新排序之后就是: [-1, 1, 3, 4, 6, 9]
但是在存储时是将排序后数值对应于x数组的下标存储在y中,所以:
y = [3, 0, 2, 1, 4, 5]

猜你喜欢

转载自blog.csdn.net/qq_41542989/article/details/108836566