从MatPlotLib画布获取二进制图像数据？

Question

我正在尝试从matplotlib画布中获取二进制数据,以便将其附加到电子邮件中,但我发现这样做的唯一方法是:

filename = 'image.png'
canvas.print_figure(filename)
with open(filename, 'rb') as image:
    return image.read()

我真的很想避免磁盘IO,因为之后我不需要保留文件.

Answer 1

使用StringIO对象作为文件对象,可以将其赋予print_pngcanvas函数.

from cStringIO import StringIO
sio = StringIO()
canvas.print_png(sio)
return sio.getvalue()

(如果您使用的是Python 3,请使用io.BytesIO而不是cStringIO)