代碼如下
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