假设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函数来执行这样的任务?任何建议表示赞赏!
output = np.zeros((2,4), int)
output[:, :2] = a # broadcasts (2,) to (1,2) to (2,2)
output[:, 2:] = b
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句