在二維空間中找到一個點周圍的點

Question

我在二維空間中有兩個點數組：藍色的（包含數百個點）和紅色的（包含數十萬個點）。
對於每個藍點，我必須找到圍繞它的三個最近的紅點，給定它們的坐標，如圖所示。

哪個是計算效率最高的算法？

Answer 1

為一組創建一個scipy.KDTree 。 然后對第二組使用query 。 https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.KDTree.query.html#scipy.spatial.KDTree.query

編輯：這並不能保證找到的點圍繞查詢點，但它為您提供了很好的候選者。 此時，您可以強制檢查三個點是否圍繞查詢點，如果沒有，則與更多鄰居重試。