matplotlib/examples/misc/agg_buffer_to_array.py at unpyplot · a52/matplotlib · GitHub

Repository navigation

"""

===================

Agg Buffer To Array

===================

Convert a rendered figure to its image (NumPy array) representation.

"""

import matplotlib.pyplot as plt

import numpy as np

# make an agg figure

fig, ax = plt.subplots()

ax.plot([1, 2, 3])

ax.set_title('a simple figure')

fig.canvas.draw()

# grab the pixel buffer and dump it into a numpy array

X = np.array(fig.canvas.renderer.buffer_rgba())

# now display the array X as an Axes in a new figure

fig2 = plt.figure()

ax2 = fig2.add_subplot(111, frameon=False)

ax2.imshow(X)

plt.show()