python - How to plot images in subplots

Question

Suppose I have 3 directories of .jpg files: dataset 1, dataset 2, dataset 3.

I would like to make a 5 by 3 subplots using matplotlib. For each row, the subplot shows the data from dataset 1, dataset 2 and dataset 3 in order. The expected format is like this:

plot1, plot2, plot3,

plot4.......

plot13, plot14, plot15.

How should I do that?

something like this:

plt.figure(figsize=(10, 10)) 
for data1, data2, data3 in dataset1, dataset2, dataset3"
....

Answer 1

• This example uses Path(...).glob() from pathlib to find all of the image paths in each directory, and unpack them in a list comprehension.
• matplotlib.pyplot.imread and matplotlib.pyplot.imshow are used to read and show the images, respectively.

import matplotlib.pyplot as plt
from pathlib import Path

# create a list of directories
dirs = ['../Pictures/dataset1', '../Pictures/dataset2', '../Pictures/dataset3']

# extract the image paths into a list
files = [f for dir_ in dirs for f in list(Path(dir_).glob('*.jpg'))]

# create the figure
fig, axs = plt.subplots(nrows=5, ncols=3, figsize=(10, 10))

# flatten the axis into a 1-d array to make it easier to access each axes
axs = axs.flatten()

# iterate through and enumerate the files, use i to index the axes
for i, file in enumerate(files):
    
    # read the image in
    pic = plt.imread(file)

    # add the image to the axes
    axs[i].imshow(pic)

    # add an axes title; .stem is a pathlib method to get the filename
    axs[i].set(title=file.stem)

# add a figure title
fig.suptitle('Images from https://www.heroforge.com/', fontsize=18)