python - 在 Numpy 中，从两个数组中找出每对之间的欧氏距离

Question

我有两个二维坐标点 (x,y) 数组

a = [ (x1,y1), (x2,y2), ... (xN,yN) ]
b = [ (X1,Y1), (X2,Y2), ... (XN,YN) ]

如何在 1xN 数组中找到每个对齐对 (xi,yi) 到 (Xi,Yi) 之间的欧氏距离？

scipy.spatial.cdist 函数为我提供了 NxN 数组中所有对之间的距离。

如果我只是使用norm 函数来一个一个地计算距离，它似乎很慢。

是否有内置函数可以执行此操作？

Answer 1

我没有看到内置的，但你可以很容易地自己做。

distances = (a-b)**2
distances = distances.sum(axis=-1)
distances = np.sqrt(distances)