從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畫布函數。

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

（如果您使用的是Python 3，請使用io.BytesIO而不是cStringIO ）