在二维空间中找到一个点周围的点

Question

我在二维空间中有两个点数组：蓝色的（包含数百个点）和红色的（包含数十万个点）。
对于每个蓝点，我必须找到围绕它的三个最近的红点，给定它们的坐标，如图所示。

哪个是计算效率最高的算法？

Answer 1

为一组创建一个scipy.KDTree 。 然后对第二组使用query 。 https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.KDTree.query.html#scipy.spatial.KDTree.query

编辑：这并不能保证找到的点围绕查询点，但它为您提供了很好的候选者。 此时，您可以强制检查三个点是否围绕查询点，如果没有，则与更多邻居重试。