3
如何使用matplotlib颜色图将NumPy数组转换为PIL图像
我有一个简单的问题，但找不到很好的解决方案。 我想获取一个代表灰度图像的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)