Итерировать изображения через папку

Question

У меня есть папка с полутоновыми изображениями от 0 до 9, и их около 2400, мне нужно загрузить их в python, чтобы все Zeors были объединены в массив, затем в единое целое как массив и т. Д. , Я использовал приведенный ниже код для загрузки одного изображения в виде массива, но я не знаю, как загрузить все изображения и сгруппировать каждое число вместе. я думал об итерации по папке.

Кто-нибудь знает, как это сделать, или есть другая идея?

import imageio

im = imageio.imread('Train/1.jpg')

Answer 1

Вы можете сделать это следующим образом:

import imageio
for i in range(9):
    im = imageio.imread('Train/'+str(i)+'.jpg')

Вы также можете создать трехмерную переменную, третье измерение которой равно числу файлов для чтения полных данных в одном трехмерном массиве.