复制数组行
Replicate array rows
我想将一维数组(n 维)的元素分配给二维数组(mxn 维)的每一行。有没有办法使用 numpy 广播在一行中完成?
您可以使用 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)
我想将一维数组(n 维)的元素分配给二维数组(mxn 维)的每一行。有没有办法使用 numpy 广播在一行中完成?
您可以使用 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)