将 numpy 数组扩展到 n*length 并复制其元素的有效方法？ [复制]

Question

我正在寻找一种快速获取numpy数组的方法，例如[[1,2,3,4]] 并将其转换为自身的扩展版本，其元素重复N 次。 IE。如果N = 2，那么[[1,2,3,4]] -> [[1,1,2,2,3,3,4,4]]

显然我可以使用明确的 for 来强制它，但我想知道是否可以通过某种方式对其进行矢量化以加快速度？

编辑：我使用 [[1,2,3,4]] 作为 np.array([1,2,3,4]) 的简写，抱歉造成混淆。另外，感谢那些提到 np.repeat 的人！这正是我所需要的。

Answer 1

试试这个：

from numpy import repeat

x = [[1,2,3,4]]
N = 3
y = repeat(x, N).reshape((1,-1))

print(y)

编辑： Quang 的解决方案更短，我承认...

y = repeat(x, N, axis=-1)