Matplotlib: guardar el gráfico en numpy array

Question

En Python y Matplotlib, es fácil mostrar el diagrama como una ventana emergente o guardar el trazado como un archivo PNG. ¿Cómo puedo guardar la trama en una matriz numpy en formato RGB?

Answer 1

Este es un truco útil para pruebas unitarias y similares, cuando se necesita hacer una comparación de píxel a píxel con una gráfica guardada.

Una forma es usar fig.canvas.tostring_rgb y luego numpy.fromstring con el tipo de letra correspondiente. También hay otras formas, pero esta es la que tiendo a usar.

E.g.

import matplotlib.pyplot as plt 
import numpy as np 

# Make a random plot... 
fig = plt.figure() 
fig.add_subplot(111) 

# If we haven't already shown or saved the plot, then we need to 
# draw the figure first... 
fig.canvas.draw() 

# Now we can save it to a numpy array. 
data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='') 
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))