我有一个简单的问题,但我找不到解决它的好办法.
我想获取一个表示灰度图像的NumPy 2D数组,并在应用一些matplotlib colored颜色 映射时将其转换为RGB PIL图像.
我可以使用pyplot.figure.figimage
命令获得合理的PNG输出:
dpi = 100.0
w, h = myarray.shape[1]/dpi, myarray.shape[0]/dpi
fig = plt.figure(figsize=(w,h), dpi=dpi)
fig.figimage(sub, cmap=cm.gist_earth)
plt.savefig('out.png')
虽然我可以调整它以获得我想要的(可能使用StringIO来获得PIL图像),但我想知道是否有更简单的方法来实现这一点,因为这似乎是图像可视化的一个非常自然的问题.比如说,类似这样的事情:
colored_PIL_image = magic_function(array, cmap)