Powracający obraz Matplotlib jako ciąg

Question

Używam matplotlib w aplikacji django i chcę bezpośrednio zwrócić renderowany obraz. Do tej pory mogę przejść plt.savefig(...), a następnie zwrócić lokalizację obrazu.

Co chcę zrobić, to:

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

Jakieś pomysły?

Answer 1

Django HttpResponse obiekt obsługuje plikopodobny API i można przekazać plik-Object do savefig.

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

W związku z tym można zwrócić obraz bezpośrednio w numerze HttpResponse.

Answer 2

Zatrudniamy Duck Typing i przekazać obiekt z własną rękę, w przebraniu obiektu pliku

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

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

można użyć myfile = StringIO.StringIO() zamiast w prawdziwym kodzie i zwraca dane w réponse np

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

Answer 3

co z 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

Istnieje przepis w Matplotlib Cookbook, który robi dokładnie to. Zasadniczo wygląda to następująco:

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 

Umieść to w swoim pliku poglądów, wskaż adres URL i uruchom.

Edycja: Jak wspomniano, jest to uproszczona wersja przepisu w książce kucharskiej. Jednak wygląda na to, że istnieje różnica między wywołaniem print_png i savefig, przynajmniej w początkowym teście, który zrobiłem. Dzwonienie pod numer fig.savefig(response, format='png') dało obraz, który był większy i miał białe tło, podczas gdy oryginał canvas.print_png(response) dał nieco mniejszy obraz z szarym tłem. Tak, chciałbym zastąpić kilka ostatnich linii powyżej:

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

Nadal trzeba mieć utworzony materiał canvas.