【全新版本】matplotlib figure转为numpy array或者PIL图像进行显示

简述

之前在网上看到了一个版本，但是写得太冗长了

因为他的思路是

1. plt
2. np
3. PIL

实际上反过来操作，会更加简单

新版本

• 下面版本会把plt.plot的结果转成PIL Image的格式。
• PIL转numpy非常简单np.asarray(Image) 即可

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

def f(a):
	tmpfile = BytesIO()
	plt.plot(a,a)
	plt.savefig(tmpfile)
	return Image.open(tmpfile)