我有一個n維向量,並且我想使用euclidian距離在n維向量列表中找到它的k個最近鄰居。使用歐幾里得距離在numpy數組列表中找到numpy數組的最近鄰居
我寫了下面的代碼(k = 10),它運行得很慢,但我想知道是否有更優化的解決方案。
def nearest_neighbors(value, array, nbr_neighbors=1):
return np.argsort(np.array([np.linalg.norm(value-x) for x in array]))[:nbr_neighbors]
你應該選擇他的答案的答案,將其添加到您的文章作爲最終編輯。 – dawg