Составьте список из numpy массивов

Question

У меня есть список numpy массивов следующих форм:

(16, 250, 2)
(7, 250, 2)
(1, 250, 2)

Я пытаюсь сложить их все вместе, чтобы они были одним массивом numpy формы:

(24, 250, 2)

Я пробовал использовать np.stack, но получаю сообщение об ошибке:

ValueError: all input arrays must have the same shape

Answer 1

Хитрость заключается в том, чтобы использовать правильный метод наложения, в вашем случае, поскольку вы складываете вертикально, вы должны использовать np.vstack

import numpy as np

a = np.random.random((16, 250, 2))
b = np.random.random((7, 250, 2))
c = np.random.random((1, 250, 2))

arr = np.vstack((a,b,c))
arr.shape
(24, 250, 2)

Answer 2

Вы можете использовать np.concatenate: Вы получите это: -

a = np.random.rand(16,250,2)
b = np.random.rand(7,250,2)
c = np.random.rand(1,250,2)
print(np.shape(np.concatenate([a,b,c], axis=0))

Вывод:

(24,250,2)