subplot.py

%matplotlib inline
import matplotlib.pyplot as plt

def sxs(images, idxs=[]):
    plt.figure(figsize=(100, 100))
    assert idxs, 'define me'
    for i, idx in enumerate(idxs):
        plt.subplot(1, len(idxs), i+1)
        plt.imshow(images[idx], cmap='gray')

    plt.tight_layout()
    plt.show()