如何把matplotlib figure转为numpy array或者PIL图像

国外网友的一个回答,链接如下,质量不知道比国内网友那些无脑转载高到哪里去了.

How to convert matplotlib figure to numpy array or PIL image?

里面有源码,可以直接下载尝试,亲测可用.需要修改的地方如下: 

import Image

改为

from PIL import Image

因为python3中没有Image这个库,需要导入PIL中的Image模块.

函数fig2img中的:

return Image.fromstring("RGBA", (w,h), buf.tostring())

改为

return Image.frombytes("RGBA", (w,h), buf.tostring())

python3中fromstring已被弃用,所以改为frombytes.