Python укладывает изображения в куб

Question

Есть ли быстрый способ укладывать изображения в куб?Я знаю, что вы можете использовать np.append для добавления двух, но если их много, вам нужен forloop и rehape.Есть ли более разумные способы сделать это?

Answer 1

Вы можете использовать np.stack, который принимает произвольное количество массивов и объединяет их вдоль новой оси.

Пример:

images = [np.random.randn(8, 12) for _ in range(50)]
stacked =  np.stack(images, axis=0)
print(stacked.shape)  # output: (50, 8, 12)