scipy空间kdtree查找查询点给定距离内的所有点

Question

我在R ³ 中设置了一个数据，并希望找到距查询点有一段距离的所有点。这是一个示例，其中我的数据点是[0,0,0] ... [4,4,4]，查询点是[1,1,1.1]，距离是40。我正在尝试遵循这些文档：

https://docs.scipy.org/doc/scipy/reference/generated/scipy.spatial.KDTree.query.html

import numpy as np
from scipy import spatial

data = [[x,x,x] for x in range(5)]
tree = spatial.KDTree(data)
print(tree.query([1,1,1.1],eps=40))

因为我将eps设置得很大，所以我希望应该返回所有点，因为它们都在给定范围内，但是事实并非如此。我还尝试了以下方法：

import numpy as np
from scipy import spatial

data = [[x,x,x] for x in range(5)]
tree = spatial.KDTree(data)
print(tree.query([1,1,1.1],distance_upper_bound=40))

但是，这给出了相同的结果。这是我得到的输出：

('0.10000000000000009', 1)

我想知道如何正确执行此操作。还有我在文档中误会了什么？我该如何以不同的方式处理文档，以便从文档中获取正确的信息？