擅长:python、mysql、java
<p>您想在这里使用<a href="http://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KDTree.html#sklearn.neighbors.KDTree.query_radius" rel="nofollow noreferrer">query_radius</a>。在</p>
<blockquote>
<p>query_radius(self, X, r, count_only = False):</p>
<p>query the tree for neighbors within a radius r</p>
<p>...</p>
</blockquote>
<p>以上链接中的示例:</p>
<pre><code>import numpy as np
np.random.seed(0)
X = np.random.random((10, 3)) # 10 points in 3 dimensions
tree = BinaryTree(X, leaf_size=2)
print(tree.query_radius(X[0], r=0.3, count_only=True))
ind = tree.query_radius(X[0], r=0.3)
print(ind) # indices of neighbors within distance 0.3
</code></pre>