可以在python中的sklearn做最近邻搜索球内的点吗？

Question

我知道sklearn可以很好地计算大数据集的k-最近邻居，但似乎它们没有搜索查询点的给定半径内的所有点的函数。

是否可以进行此类范围查询？我正在寻找sklearn类似于scipy.spatial的KDTree方法，名为query_ball_point

Answer 1

这可以通过检索1，2，4，8，...最近的邻居直到超过半径，然后丢弃离查询点太远的点来实现。