Merge pull request #14797 from tacaswell/doc_agg_buffer_example

dstansby · web-flow · commit bc76370a8ce9 · 2019-08-13T17:51:49.000+01:00
DOC: create a Agg figure without pyplot in buffer example
diff --git a/examples/misc/agg_buffer_to_array.py b/examples/misc/agg_buffer_to_array.py
@@ -7,15 +7,21 @@
 """
 import matplotlib.pyplot as plt
 import numpy as np
+from matplotlib.figure import Figure
+from matplotlib.backends.backend_agg import FigureCanvas
 
-# make an agg figure
-fig, ax = plt.subplots()
+# Create a figure that pyplot does not know about.
+fig = Figure()
+# attach a non-interactive Agg canvas to the figure
+# (as a side-effect of the ``__init__``)
+canvas = FigureCanvas(fig)
+ax = fig.subplots()
 ax.plot([1, 2, 3])
 ax.set_title('a simple figure')
-fig.canvas.draw()
-
+# Force a draw so we can grab the pixel buffer
+canvas.draw()
 # grab the pixel buffer and dump it into a numpy array
-X = np.array(fig.canvas.renderer.buffer_rgba())
+X = np.array(canvas.renderer.buffer_rgba())
 
 # now display the array X as an Axes in a new figure
 fig2 = plt.figure()
