python - python 中的 dsearchn 等价物

Question

是否有 Scipy 或 Numpy 函数可以在 python 中完成 dsearchn MATLAB 命令的工作？

dsearchn 返回给定向量中最接近输入值的索引。

Answer 1

效率差不多的方法大概是scipy的KDTree或者更好cKDTree :

from scipy.spatial import KDTree

kdt = KDTree(P.T)
kdt.query(PQ.T)

这里的P和PQ是来自dsearch的点和查询点文档。 MATLAB 使用第一个维度作为点的维度，而 scipy 使用最后一个维度，因此是转置。

不像 MATLAB 方法，它是一步完成的，使用 KDTree 我们分两步完成:首先构建树对象，然后运行 ​​query。在上面。