Numpy 将二维 arrays 堆叠到一个空的 3d 数组中

Question

from PIL import Image

train_data = np.array([])

for name in train_images_names[:100]:
    image = np.array(Image.open(name))
    train_data = np.append(train_data, image)

train_images_names 是图像文件名称的列表，每个图像有 (256, 256) 个维度。

我想制作一个 3d 数组，其中包含一百张图像 arrays（256、256、100）。

任何帮助，将不胜感激

Answer 1

尝试stack ：

np.stack([np.array(Image.open(name) for name in train_images_names[:100]],
         axis=-1)