Matplotlib: enregistrer un tracé dans un tableau numpy

Question

En Python et Matplotlib, il est facile d'afficher le tracé sous forme de fenêtre contextuelle ou de sauvegarder le tracé en tant que fichier PNG. Comment puis-je sauvegarder l'intrigue dans un tableau numérique au format RVB?

Answer 1

Ceci est une astuce pratique pour les tests unitaires et autres, lorsque vous avez besoin de faire une comparaison de pixel à pixel avec un tracé sauvegardé.

Une façon est d'utiliser fig.canvas.tostring_rgb puis numpy.fromstring avec le dtype approprié. Il y a aussi d'autres moyens, mais c'est celui que j'ai tendance à utiliser.

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,))