web-dev-qa-db-ja.com

PIL画像に読み込むことができるインメモリファイルにpylabフィギュアを保存する方法？

pILは初めてですが、すぐに解決策を見つけたいと考えています。以下は、私の最初のショットです。

import cStringIO
import pylab
from PIL import Image
pylab.figure()
pylab.plot([1,2])
pylab.title("test")
buffer = cStringIO.StringIO()
pylab.savefig(buffer, format='png')
im = Image.open(buffer.read())
buffer.close()

エラーは言う、

Traceback (most recent call last):
  File "try.py", line 10, in <module>
    im = Image.open(buffer.read())
  File "/awesomepath/python2.7/site-packages/PIL/Image.py", line 1952, in open
    fp = __builtin__.open(fp, "rb")

何か案は？ソリューションに余分なパッケージを含めたくありません。

pythonpython-imaging-librarymatplotlib
51
nye17

buf.seek(0)を呼び出すことを忘れないでください。したがって、Image.open(buf)はbufの先頭から読み取りを開始します。

import io
from PIL import Image
import matplotlib.pyplot as plt

plt.figure()
plt.plot([1, 2])
plt.title("test")
buf = io.BytesIO()
plt.savefig(buf, format='png')
buf.seek(0)
im = Image.open(buf)
im.show()
buf.close()
78
unutbu