使用scipy查找两个阵列点之间的最短距离

Question

我有两个数组centroids和nodes

我需要找到centroids中每个点到nodes中任何点的最短距离

centroids的输出如下

array([[12.52512263, 55.78940022],
       [12.52027731, 55.7893347 ],
       [12.51987146, 55.78855611]])
       

nodes的输出如下

array([[12.5217378, 55.7799275],
       [12.5122589, 55.7811443],
       [12.5241664, 55.7843297],
       [12.5189395, 55.7802709]])

我使用以下代码获得最短距离

shortdist_from_centroid_to_node = np.min(cdist(centroids,nodes))

然而，这是我得到的输出（我应该得到3行输出）

Out[131]: 3.0575613850140956e-05

有人能说明问题出在哪里吗？谢谢

Answer 1

我想您需要的只是添加一个名为axis的arg，如下所示：

shortdist_from_centroid_to_node = np.min(cdist(centroids,nodes), axis=1)

至于轴arg的含义，您可以参考numpy.min。总而言之，您需要在每一行上使用最小值，而不是在整个矩阵上使用最小值