matplotlib

Travis-CI:

Related Topics

Documentation overview
- Matplotlib Examples
  - pylab_examples Examples
    - Previous: pylab_examples example code: accented_text.py
    - Next: pylab_examples example code: agg_buffer_to_array.py

This Page

pylab_examples example code: agg_buffer.py¶

(Source code, png, pdf)

../../_images/agg_buffer.png

"""
Use backend agg to access the figure canvas as an RGB string and then
convert it to an array and pass it to Pillow for rendering.
"""

import numpy as np

import matplotlib.pyplot as plt
from matplotlib.backends.backend_agg import FigureCanvasAgg


try:
    from PIL import Image
except ImportError:
    raise SystemExit("Pillow must be installed to run this example")

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

canvas = plt.get_current_fig_manager().canvas

agg = canvas.switch_backends(FigureCanvasAgg)
agg.draw()
s = agg.tostring_rgb()

# get the width and the height to resize the matrix
l, b, w, h = agg.figure.bbox.bounds
w, h = int(w), int(h)

X = np.fromstring(s, np.uint8)
X.shape = h, w, 3

try:
    im = Image.fromstring("RGB", (w, h), s)
except Exception:
    im = Image.frombytes("RGB", (w, h), s)

# Uncomment this line to display the image using ImageMagick's
# `display` tool.
# im.show()

Keywords: python, matplotlib, pylab, example, codex (see Search examples)