Как построить несколько изображений из файла в Python?

Question

Я пытаюсь нанести несколько изображений из файла в блокнот jupyter. Изображения отображаются, но они в одном столбце. Я использую:

%matplotlib inline
from os import listdir
form PIL import image as PImage
from matplotlib import pyplot as plt

def loadImages(path):
    imagesList = listdir(path)
    loadedImages = []
    for image in imagesList:
        img = PImage.open(path+image)
        LoadedImages.append(img)
        Return loadedImages
path = 'C:/Users/Asus-PC/flowers/'
imgs = loadImages(path)

for img in imgs
    plt.figure()
    plt.imshow(img)

Я бы хотел, чтобы они отображались в сетке (строка и столбец). Частично проблема в том, что я не знаю, что означают аргументы add_subplot. Как мне этого добиться?

Answer 1

Вы можете создать несколько субплотов, используя plt.subplots. Определите количество столбцов и строк по количеству загруженных изображений. Что-то вроде:

from os import listdir
import matplotlib.pyplot as plt

path = 'C:/Users/Asus-PC/flowers/'
imagesList = listdir(path)
n_images = len(imagesList)    

figure, axes = plt.subplots(n_images, 1)   # (columns, rows)    
for ax, imgname in zip(axes, imagesList):  # Iterate over images and
        img = plt.imread(path+imgname)     # axes to plot one image on each.
        ax.imshow(img)                     # You can append them to an empty
                                           # list if you need them later.
plt.show()