我正在使用以下示例:
from scipy import spatial
x, y = np.mgrid[0:5, 2:8]
tree = spatial.KDTree(list(zip(x.ravel(), y.ravel())))
pts = np.array([[0, 0], [2.1, 2.9]])
idx = tree.query(pts)[1]
data = tree.data[??????????]
如果我输入两个任意点(请参见变量
pts
),我希望返回所有坐标对,这些坐标在由这两个点定义的矩形内(KDTree找到最近邻)。因此,在这种情况下:array([[0, 0],
[0, 1],
[0, 2],
[1, 0],
[1, 1],
[1, 2],
[2, 0],
[2, 1],
[2, 2]])
我该如何从树形数据中实现这一点?