imshow a panel of images

subimshow.py

import numpy as np
from matplotlib import pyplot as plt
def subimshow(image, title=''):
    """
    Show each band seperately.
    
    The shape si a square or rectangle of images
    
    image: array of shape (channel, height, width)
    
    """
    nb_bands = image.shape[0]
    ncols=int(np.ceil(np.sqrt(image.shape[0])))
    nrows=int(np.ceil(nb_bands/float(ncols)))
    fig, ax = plt.subplots(nrows=nrows, ncols=ncols, sharex=True, sharey=False, figsize=(15,15))
    if title: fig.suptitle('{} shape={} min={:2.2f} mean{:2.2f} max={:2.2f} std={:2.2f} dtype={}'.format(title, image.shape, image.min(), image.mean(), image.max(), image.std(), image.dtype), fontsize=14)
    ax=ax.flatten()
    for i in range(len(image)):
        d=image[i].astype(float)
        d/=d.sum()
        ax[i].imshow(d)
        ax[i].set_title('Band %s'%i)
        ax[i].autoscale(False)
    
    plt.show()
    
subimshow(np.random.random((5,50,50)),'5 bands')

subimshow(np.random.random((13,50,50)), '13 bands')