复制阵列行

Question

我想将1d数组(维度n(的元素分配到2d数组(维度mxn(的每一行中。有没有一种方法可以使用numpy广播在一行中完成？

Answer 1

IIUC您可以使用np.broadcast_to。这里有一个例子：

a = np.array([1,3,4,2])
m = 5
np.broadcast_to(a,(m,len(a)))
array([[1, 3, 4, 2],
[1, 3, 4, 2],
[1, 3, 4, 2],
[1, 3, 4, 2],
[1, 3, 4, 2]])

或者另一种方法可以是使用np.vstack:

np.vstack([a]*m)