Figure to image as a numpy array

by

In order to get the figure contents as RGB pixel values, the matplotlib.backend_bases.Renderer needs to first draw the contents of the canvas.

You can do this by manually calling canvas.draw():

from matplotlib.figure import Figure

fig = Figure()
canvas = fig.canvas
ax = fig.gca()

ax.text(0.0,0.0,"Test", fontsize=45)
ax.axis('off')

canvas.draw()  # Draw the canvas, cache the renderer

image_flat = np.frombuffer(canvas.tostring_rgb(), dtype="uint8")  # (H * W * 3,)
# NOTE: reversed converts (W, H) from get_width_height to (H, W)
image = image_flat.reshape(*reversed(canvas.get_width_height()), 3)  # (H, W, 3)

See here for more info on the Matplotlib API.

More Related Contents:

  1. Plot a function in python
  2. ImportError: DLL load failed: The specified module could not be found
  3. understanding matplotlib.subplots python [duplicate]
  4. Plotting a 2D heatmap with Matplotlib
  5. A tool to convert MATLAB code to Python [closed]
  6. How to convert a NumPy array to PIL image applying matplotlib colormap
  7. How can I attach a pyplot function to a figure instance?
  8. matplotlib – extracting data from contour lines
  9. Fitting a Normal distribution to 1D data
  10. Matplotlib figure to image as a numpy array
  11. how to handle an asymptote/discontinuity with Matplotlib
  12. Linear regression with matplotlib / numpy
  13. Problem in combining bar plot and line plot (python)
  14. Matplotlib Plot Lines with Colors Through Colormap
  15. Annotate Time Series plot in Matplotlib
  16. Matplotlib runs out of memory when plotting in a loop
  17. ‘invalid value encountered in double_scalars’ warning, possibly numpy
  18. Saving Matplotlib graphs to image as full screen
  19. How do I plot list of tuples in Python?
  20. plotting a histogram on a Log scale with Matplotlib
  21. How to install matplotlib with Python3.2
  22. Plotting implicit equations in 3d
  23. Cannot get minor grid lines to appear in matplotlib figure
  24. Adding colors to a 3d quiver plot in matplotlib
  25. How to have logarithmic bins in a Python histogram
  26. How to shade region under the curve in matplotlib
  27. Extract csv file specific columns to list in Python
  28. 3D Plots over non-rectangular domain
  29. Annotate Time Series plot
  30. ValueError: x and y must be the same size

Leave a Comment