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)