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

Question

假设 numpy 向量 和矩阵 b 如下:

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

我想连接向量 进入 矩阵 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