Trả lại hình ảnh Matplotlib thành chuỗi

Question

Tôi đang sử dụng matplotlib trong ứng dụng django và muốn trực tiếp trả về hình ảnh được hiển thị. Cho đến nay tôi có thể đi plt.savefig(...), sau đó trả lại vị trí của hình ảnh.

Những gì tôi muốn làm là:

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

Bất kỳ ý tưởng?

Answer 1

Đối tượng của Django là đối tượng tập tin để lưu.

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

Do đó, bạn có thể trả lại hình ảnh trực tiếp trong HttpResponse.

Answer 2

dụng ducktyping và vượt qua một đối tượng của riêng bạn, trong ngụy trang của đối tượng tập tin

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

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

bạn có thể sử dụng myfile = StringIO.StringIO() thay vì trong mã sản và trả lại dữ liệu trong phản ứng ví dụ

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

Answer 3

khoảng 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

Có công thức trong Matplotlib Cookbook thực hiện chính xác việc này. Ở cốt lõi của nó, có vẻ như:

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 

Đặt điều đó vào tệp lượt xem, trỏ URL của bạn đến đó và bạn đang tắt và chạy.

Chỉnh sửa: Như đã lưu ý, đây là phiên bản đơn giản của công thức trong sách dạy nấu ăn. Tuy nhiên, có vẻ như có sự khác biệt giữa việc gọi số print_png và savefig, ít nhất là trong thử nghiệm ban đầu mà tôi đã làm. Gọi số fig.savefig(response, format='png') cho hình ảnh lớn hơn và có nền trắng, trong khi hình ảnh gốc canvas.print_png(response) đã cho hình ảnh hơi nhỏ hơn với nền màu xám. Vì vậy, tôi sẽ thay thế một vài dòng cuối cùng ở trên bằng:

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

Bạn vẫn cần phải có canvas khởi tạo.