如何将向量连接到 numpy 矩阵的行中？

Question

假设 numpy 向量a和矩阵b如下：

import numpy as np
a = np.array([1,2])
b = np.array([[3,4],[5,6]])

我想将向量a连接到矩阵 b 的每一行中。预期输出如下：

output=np.array([[1,2,3,4],[1,2,5,6]])

我有一个工作代码如下：

output=np.array([np.concatenate((a,row)) for row in b] )

有没有更快的 numpy 函数来执行这样的任务？任何建议表示赞赏！

Answer 1

output = np.zeros((2,4), int)
output[:, :2] = a    # broadcasts (2,) to (1,2) to (2,2)
output[:, 2:] = b