python - 在 3d 空间中找到 python numpy.ndarray 的最近邻居

Question

我有一个 3d 点的 numpy.ndarray，即它的 np.shape 是 (4350,3) 和 np.shape (10510,3) 的 3d 点的第二个 numpy.ndarray。现在我正在尝试找到合适的 python 包来尽快计算第一个数组中点的第二个数组中的最近邻居。

我在这里发现了一个非常相似的问题:find the k nearest neighbours of a point in 3d space with python numpy但我不明白如何使用那里的解决方案来解决我的问题。

我非常非常感谢您对此提供的帮助!

Answer 1

这是 KDTree 方式:

from scipy.spatial import KDTree

data= np.random.rand(10510,3)
sample= np.random.rand(4350,3)
kdtree=KDTree(data)

然后 dist,points=kdtree.query(sample,2) 将在大约一秒钟内为您提供 4350 个候选人的 2 个最佳邻居。