如何计算numpy数组中两行之间的欧氏距离

Question

我有一个numpy数组，如：

import numpy as np
a = np.array([[1,0,1,0],
             [1,1,0,0],
             [1,0,1,0],
             [0,0,1,1]])

我想计算每对行之间的euclidian distance。

from scipy.spatial import distance
for i in range(0,a.shape[0]):
    d = [np.sqrt(np.sum((a[i]-a[j])**2)) for j in range(i+1,a.shape[0])]
    print(d)

[1.4142135623730951, 0.0, 1.4142135623730951]

[1.4142135623730951, 2.0]

[1.4142135623730951]

[]

既然我必须在一个巨大的numpy数组上运行这段代码，有没有更好的pythonic方法可以做到这一点？

Answer 1

我将itertools.combinations与差分向量的np.linalg.norm一起使用（这是欧几里德距离）：

import numpy as np
import itertools
a = np.array([[1,0,1,0],
              [1,1,0,0],
              [1,0,1,0],
              [0,0,1,1]])

print([np.linalg.norm(x[0]-x[1]) for x in itertools.combinations(a, 2)])

为了理解，请从docs：
combinations('ABCD', 2)给出AB AC AD BC BD CD。在你的例子中，A，B，C和D是矩阵的行a，所以上面代码中出现的x[0]-x[1]是a行中向量的差分向量。