代码如下
distances = sqDistances**0.5
print("距离和开方",distances)
sortedDistIndices = distances.argsort() #argsort函数返回的是数组值从小到大的索引值
print("从小到大的距离",sortedDistIndices)
运行结果如下
距离和开方 [9.6 1.99248588 7.6 2.03224014 2.63248932]
从小到大的距离 [1 3 4 2 0]
argsort的作用是对列表内的数据升序排列,并且给出按升序排列的数据在列表中的位置,如:1.99最小,它在原列表中的位置是1
9.6最大,在原列表的位置是0