首页 > 解决方案 > 在 Numpy 中找到最近点的最快方法

问题描述

我有一个 Numpya形状数组,5000 * 3其中包含 3D 空间中 5000 个点的坐标。我还有另一个b形状数组,512 * 3它再次包含该 3D 空间中另外 512 个点的坐标。我希望从中选择a最接近点的512 个点b。这意味着要从 中的点中找到一个c形状的 Numpy 数组,这样对于每个,是最接近 的点,来自 中的所有 5000 个点。512 * 3a1<=i<=512c[i]b[i]a

要查找c,有一种方法可以逐点比较并找到最接近的,但速度很慢。我希望找到另一种更快的方法来解决这个问题。

标签: pythonnumpy

解决方案


推荐阅读