Iterate through a series of matplotlib plots, waiting for keyboard input (from the terminal)

many_plots.py

import numpy as np
from matplotlib import pyplot as plt
plt.ion()

# say you have a bunch of distributions you'd like to look at
data_sets = [np.random.poisson(lam, size=100), for lam in np.random.randint(1, 100, size=10)]

# in a REPL, this one-liner makes it easy to plot each one successively waiting for ENTER from the terminal
_ = [[plt.hist(data), input()] for data in data_sets]