Matplotlib context manager helpers

matplotlib_contextmanagers.py

import matplotlib.pyplot as plt
import contextlib

@contextlib.contextmanager
def subplots(*args, **kwargs):
    fig, axes = plt.subplots(*args, **kwargs)
    try:
        yield fig, axes
    finally:
        plt.show()
        plt.close(fig)