Повторите / продублируйте данный массив numy десять раз

Question

У меня фигура array, (320, 320, 3). Я хочу повторить / дублировать эти данные 10 раз, и хочу получить новый массив формы ( 10 , 320, 320, 3).

Как это сделать?

array = np.ones((320, 320, 3))
print (array.shape)
(320, 320, 3)

Я пробовал как:

res = np.tile(array, 10)
print (res.shape)

(320, 320, 30).

Но я хочу shape из,

(10, 320, 320, 3)

Answer 1

Вы можете использовать np.resize, который будет выкладываться, если новый размер больше старого:

array = np.ones((320, 320, 3))

new_array = np.resize(array, (10, *array.shape))
print(new_array.shape)
# (10, 320, 320, 3)

С Документы :

numpy.resize(a, new_shape): если новый массив больше исходного массива, то новый массив заполняется повторными копиями.

Answer 2

res = np.tile(array, (10,1,1,1))
print (res.shape)

Answer 3

Мы можем использовать np.broadcast_to -

np.broadcast_to(a,(10,)+a.shape).copy() # a is input array

Если у нас все в порядке с представлением, пропустите .copy(), чтобы получить практически бесплатное время выполнения и нулевые накладные расходы памяти.

Мы также можем использовать np.repeat -

np.repeat(a[None],10,axis=0)