matplotlib pdf savefig exiting early

Question

So I've copied the example given here and when I run it I get:

Traceback (most recent call last):
  File "C:\Users\User\Documents\Project work\pdf.py", line 9, in <module>
    with PdfPages('multipage_pdf.pdf') as pdf:
AttributeError: __exit__

So where do I go from here? Thanks

Answer 1

PdfPages has become a context manager only in version 1.3.1. See the chagelog .

In particular, observe the following line:

Added a context manager for creating multi-page pdfs (see matplotlib.backends.backend_pdf.PdfPages ).

Answer 2

In addition to the accepted answer, I had to do this:

import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages

And also avoid using plt.figure() which invokes the display.