2d numpy数组的矢量化差异

时间:2017-09-24 12:35:49

标签: python arrays numpy vectorization numpy-broadcasting

我有两个numpy数组A(500,50)和B(100,50),我想计算A的每一行与B的所有行之间的差异,例如dist(i,j)是第i行之间的差异B的A和第j行。我希望使用广播进行矢量化实现并且没有循环。

下面给出了使用两个循环的实现:

for i in range(A.shape[0]):
    for j in range(B.shape[0])
        dist[i][j]=A[i]-B[j]

0 个答案:

没有答案