[Python] Plot Mosaic

[Python] Plot Mosaic

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import ImageGrid

def plot_mosaic(images, titles, n_rows=2):
    
    fig = plt.figure(figsize=(16, 16.))
    grid = ImageGrid(fig, 111,
                     nrows_ncols=(n_rows, len(images) // n_rows),  # creates 2x2 grid of axes
                     axes_pad=0.3,  # pad between axes in inch.
                     )

    for ix, (ax, im) in enumerate(zip(grid, np.squeeze(images))):
        # Iterating over the grid returns the Axes.
        if titles and titles[ix]:
            ax.set_title(f'{titles[ix]}')
        ax.imshow(im[:,:,::-1])

    plt.show()