python - 将 numpy 数组的每一行与第二个 numpy 数组的所有行相加(python)

Question

我想知道是否有任何快速方法可以将第一个数组的每一行与第二个数组的所有行相加。在这种情况下，两个数组的列数相同。例如，如果 array1.shape = (n,c) 和 array2.shape = (m,c)，生成的数组将是 array3.shape = ((n*m), c)

看下面的例子:

array1 = np.array([[0, 1, 2],
                   [3, 4, 5],
                   [6, 7, 8]])

array2 = np.array([[0, 1, 2],
                   [3, 4, 5]])

结果是:

array3 = np.array([[0,  2,  4],
                   [3,  5,  7]
                   [3,  5,  7]
                   [6,  8, 10]
                   [6,  8, 10]
                   [9, 11, 13]])

我认为我能做到这一点的唯一方法是将一个数组的每一行重复另一个数组的行数。例如，通过执行 np.repeat(array1, len(array2), axis=0) 然后将该数组与 array2 相加。但是，如果行数太大，这不是很实用。另一种方法是使用 for 循环，但这太慢了。

还有其他更好的方法吗？

提前致谢。

Answer 1

将 array1 扩展到 3D 以便它可以针对 2D array2 进行广播，然后执行广播加法并且需要最终 reshape 以获得所需的输出-

In [30]: (array1[:,None,:] + array2).reshape(-1,array1.shape[1])
Out[30]: 
array([[ 0,  2,  4],
       [ 3,  5,  7],
       [ 3,  5,  7],
       [ 6,  8, 10],
       [ 6,  8, 10],
       [ 9, 11, 13]])