Scipy cKDTree最近接のゼロを含む距離

Question

Skipy cKDTreeを使用して遠方にある隣人を探したい。最も重要なのは、ポイント自体（距離ゼロ）も欲しいということです。 cKDTree。クエリーはすべてのネイバーを与えますが、距離はゼロです。

これを行う方法はありますか？

おかげ

Answer 1

私は本当にあなたの問題を再現することはできません（またはそれはあなたがツリーを照会するために使用されている方法に依存かもしれません）。返し

>>> idx = tree.query_ball_point((4, 4), 2) 
>>> points_ref[idx] 
# array([[3, 3], [4, 4], [5, 4]]) 

：ような何かを与えることができます

>>> from scipy.spatial import cKDTree 
>>> import numpy as np 

>>> points_ref = np.array([(1, 1), (3, 3), (4, 4), (5, 4), (6, 6)]) 
>>> tree = cKDTree(points_ref) 

方法cKDTree.query_ball_pointでポイント(4, 4)周り2の距離で最も近い隣人のために照会：このコードの簡単なスニペットを考慮

距離が0の点。

n-また、0の距離でポイントを返すように思わcKDTree.query方法と最寄り隣人：

>>> _, idx = tree.query((3, 3), k=2) 
>>> points_ref[idx] 
# array([[3, 3], [4, 4]])