Bracketing Pattern (Context Manager) for Matplotlib Figures #python #matplotlib

mpl_figure_contextmanager.py

import contextlib
import matplotlib.pyplot as plt

@contextlib.contextmanager
def figure(*args, **kwargs):
    fig = plt.figure(*args, **kwargs)
    yield fig
    plt.close(fig)

# use it like `with figure() as fig:`