以字符串形式返回 Matplotlib 图像

Question

我在 django 应用程序中使用 matplotlib，想直接返回渲染的图像。 到目前为止，我可以转到 plt.savefig(...)，然后返回图像的位置。

我想做的是：

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

有什么想法吗？

Answer 1

Django 的 HttpResponse 对象支持类似文件的 API，您可以将文件对象传递给 savefig。

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

因此，您可以直接在 HttpResponse 中返回图像。

Answer 2

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 3

Matplotlib Cookbook 中有一个食谱可以做到这一点。 从本质上讲，它看起来像：

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

将其放入视图文件中，将 URL 指向它，然后就可以开始运行了。

编辑：如上所述，这是食谱中食谱的简化版本。 但是，至少在我所做的初始测试中，调用 print_png 和 savefig 之间似乎存在差异。 调用 fig.savefig(response, format='png') 得到的图像更大且具有白色背景，而原始的 canvas.print_png(response) 给出带有灰色背景的稍小图像。 因此，我将上面的最后几行替换为：

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

不过，您仍然需要实例化画布。

Answer 4

使用 ducktyping 并传递您自己的对象，以文件对象的形式伪装，

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

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

您可以在实际代码中使用 myfile = StringIO.StringIO() 代替，并在响应中返回数据，例如

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