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

Question

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

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

我真的想避免使用磁盘IO，因为之后我不需要保留文件。

Answer 1

使用StringIO对象作为文件对象，可以将其赋予print_png canvas函数。

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

（如果您使用的是Python 3，请使用io.BytesIO代替cStringIO）