Devolución de la imagen Matplotlib como cadena

Question

Estoy usando matplotlib en una aplicación django y me gustaría devolver directamente la imagen renderizada. Hasta ahora puedo ir a plt.savefig(...), luego devolver la ubicación de la imagen.

Lo que quiero hacer es:

return HttpResponse(plt.renderfig(...), mimetype="image/png") 

¿Alguna idea?

Answer 1

HttpResponse objeto de Django es compatible con la API de tipo fichero y se puede pasar un objeto de archivo de savefig.

response = HttpResponse(mimetype="image/png") 
# create your image as usual, e.g. pylab.plot(...) 
pylab.savefig(response, format="png") 
return response 

Por lo tanto, puede devolver la imagen directamente en el HttpResponse.

Answer 2

Duck Typing emplear y pasar un objeto de su cuenta, en el encubrimiento del objeto de archivo

class MyFile(object): 
    def __init__(self): 
     self._data = "" 
    def write(self, data): 
     self._data += data 

myfile = MyFile() 
fig.savefig(myfile) 
print myfile._data 

puede utilizar mi_archivo = StringIO.StringIO() en lugar de código real y devolver datos en Reponse por ejemplo,

output = StringIO.StringIO() 
fig.savefig(output) 
contents = output.getvalue() 
return HttpResponse(contents , mimetype="image/png") 

Answer 3

¿qué hay de cStringIO?

import pylab 
import cStringIO 
pylab.plot([3,7,2,1]) 
output = cStringIO.StringIO() 
pylab.savefig('test.png', dpi=75) 
pylab.savefig(output, dpi=75) 
print output.getvalue() == open('test.png', 'rb').read() # True 

Answer 4

Hay una receta en el Matplotlib Cookbook que hace exactamente esto. En su núcleo, se ve así:

def simple(request): 
    from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas 
    from matplotlib.figure import Figure 

    fig=Figure() 
    ax=fig.add_subplot(111) 
    ax.plot(range(10), range(10), '-') 
    canvas=FigureCanvas(fig) 
    response=django.http.HttpResponse(content_type='image/png') 
    canvas.print_png(response) 
    return response 

Ponlo en tu archivo de visitas, apunta tu URL hacia él, y estarás en funcionamiento.

Editar: Como se señaló, esta es una versión simplificada de una receta en el libro de cocina. Sin embargo, parece que hay una diferencia entre llamar a print_png y savefig, al menos en la prueba inicial que hice. Llamando fig.savefig(response, format='png') dio una imagen que era más grande y tenía un fondo blanco, mientras que el original canvas.print_png(response) dio una imagen ligeramente más pequeña con un fondo gris. Por lo tanto, reemplazaría las últimas líneas de arriba con:

canvas=FigureCanvas(fig) 
    response=django.http.HttpResponse(content_type='image/png') 
    fig.savefig(response, format='png') 
    return response 

Aún necesita tener el lienzo instanciado.