python - 如何将数组复制/重复 N 次到新数组中？

Question

这个问题在这里已经有了答案:





generating batch of clones from image numpy

(3 个回答)


3年前关闭。




我有:

test = np.random.randn(40,40,3)

我想做:

result = Repeat(test, 10)

所以 result包含数组 test重复 10 次，形状:

(10, 40, 40, 3)

所以创建一个带有新轴的张量来保存 test 的 10 个副本。 .我也想尽可能有效地做到这一点。我怎样才能用 Numpy 做到这一点？

Answer 1

一个可以用 np.repeat 方法连同 np.newaxis :

import numpy as np

test = np.random.randn(40,40,3)
result = np.repeat(test[np.newaxis,...], 10, axis=0)
print(result.shape)
>> (10, 40, 40, 3)