Python在PIL Image对象上保存matplotlib图

时间:2010-10-15 00:45:34

标签: python image matplotlib python-imaging-library

HI,我是否有可能从matplotlib创建一个图像并将其保存在我从PIL创建的图像对象上?听起来很难?谁能帮帮我?

2 个答案:

答案 0 :(得分:15)

在Django Framework的网页中渲染Matplotlib图像:

  • 创建matplotlib图

  • 将其另存为png文件

  • 将此图像存储在字符串缓冲区中(使用PIL)

  • 将此缓冲区传递给Django的 HttpResponse (设置 mime type image / png)

  • 返回响应对象(在这种情况下为渲染图)。

换句话说,所有这些步骤都应该放在Django view 函数中,在 views.py 中:

from matplotlib import pyplot as PLT
import numpy as NP
import StringIO
import PIL
from django.http import HttpResponse 


def display_image(request) :
    # next 5 lines just create a matplotlib plot
    t = NP.arange(-1., 1., 100)
    s = NP.sin(NP.pi*x)
    fig = PLT.figure()
    ax1 = fig.add_subplot(111)
    ax1.plot(t, s, 'b.')

    buffer = StringIO.StringIO()
    canvas = PLT.get_current_fig_manager().canvas
    canvas.draw()
    pil_image = PIL.Image.fromstring('RGB', canvas.get_width_height(), 
                 canvas.tostring_rgb())
    pil_image.save(buffer, 'PNG')
    PLT.close()
    # Django's HttpResponse reads the buffer and extracts the image
    return HttpResponse(buffer.getvalue(), mimetype='image/png')

答案 1 :(得分:3)

我有同样的问题,我偶然发现了这个答案。只是想添加上面的答案,PIL.Image.fromstring已被弃用,现在应该使用frombytes而不是fromstring。因此,我们应该修改一行:

pil_image = PIL.Image.fromstring('RGB', canvas.get_width_height(), 
                 canvas.tostring_rgb())

pil_image = PIL.Image.frombytes('RGB', canvas.get_width_height(), 
                 canvas.tostring_rgb())