Numpy广播以矢量形式执行欧氏距离

Question

我有2×4和3×4的矩阵。我想找出横排的欧几里德距离，在最后得到一个2×3矩阵。这是一个for循环的代码，它根据所有b行向量计算a中每行向量的欧几里德距离。在不使用for循环的情况下如何执行相同的操作？

 import numpy as np
a = np.array([[1,1,1,1],[2,2,2,2]])
b = np.array([[1,2,3,4],[1,1,1,1],[1,2,1,9]])
dists = np.zeros((2, 3))
for i in range(2):
      dists[i] = np.sqrt(np.sum(np.square(a[i] - b), axis=1))

Answer 1

只需在正确的位置使用np.newaxis：

 np.sqrt((np.square(a[:,np.newaxis]-b).sum(axis=2)))