我有一个简单的问题,但找不到好的解决办法。
我想要一个表示灰度图像的numpy 2D数组,并在应用一些matplotlib颜色映射时将其转换为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)
谢谢你的阅读!
接受回答中描述的方法在应用了它的评论中提到的更改之后对我也不起作用。但在下面简单的代码工作
np_数组可以是2D数组,其值为0..1 floats o2 0..255 uint8,在这种情况下,它需要cmap。对于三维阵列,将忽略cmap。
很忙的一班,但是在这里:
myarray
是用最大值1.0
进行规范化的。myarray
。0-255
范围。np.uint8()
转换为整数。Image.fromarray()
。你完成了:
使用
plt.savefig()
:使用
im.save()
:相关问题 更多 >
编程相关推荐