Show multiple PNG pictures from a list in python

Question

I would like to show the first 5 pictures from a folder, consisting of .png pictures. How can I do that, without calling them one by one via their name in python?

Thanks!

Answer 1

You can use the package os to list file in a directory. Then you filter the results to keep the files with the correct extention. You take the 5 first elements of the list. Finaly you use matplotlib.image to load the image and matplotlib.pyplot to plot the image.

import os
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

filenames = os.listdir()
filenames = list(filter(lambda x: x.endswith('.png'), filenames))
filenames = filenames[:5]

for f in filenames:
    plt.imshow(mpimg.imread(f))
    plt.show()