Skip to content

Instantly share code, notes, and snippets.

@soply
Last active January 9, 2024 14:52
Show Gist options
  • Save soply/f3eec2e79c165e39c9d540e916142ae1 to your computer and use it in GitHub Desktop.
Save soply/f3eec2e79c165e39c9d540e916142ae1 to your computer and use it in GitHub Desktop.
Plot multiple images with matplotlib in a single figure. Titles can be given optionally as second argument.
import matplotlib.pyplot as plt
import numpy as np
def show_images(images, cols = 1, titles = None):
"""Display a list of images in a single figure with matplotlib.
Parameters
---------
images: List of np.arrays compatible with plt.imshow.
cols (Default = 1): Number of columns in figure (number of rows is
set to np.ceil(n_images/float(cols))).
titles: List of titles corresponding to each image. Must have
the same length as titles.
"""
assert((titles is None)or (len(images) == len(titles)))
n_images = len(images)
if titles is None: titles = ['Image (%d)' % i for i in range(1,n_images + 1)]
fig = plt.figure()
for n, (image, title) in enumerate(zip(images, titles)):
a = fig.add_subplot(cols, np.ceil(n_images/float(cols)), n + 1)
if image.ndim == 2:
plt.gray()
plt.imshow(image)
a.set_title(title)
fig.set_size_inches(np.array(fig.get_size_inches()) * n_images)
plt.show()
@lhfy127
Copy link

lhfy127 commented Dec 9, 2023

How to save the multiple-image figure as a new image?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment