将矩阵绘制为与python中的给定单元格大小的热图

Question

我试图谷歌，但找不到任何答案。 我有一个numpy ndarray值，我想绘制为一个普通的（没有坐标轴和东西）热图，每个矩形都按给定值着色。

我想拥有它，以便一个单元格的高度为1px，宽度为5px。

最终结果是一个PNG。

有没有在python的方式来绘制这与这些单元大小参数？

我试过给定总宽度和高度的PIL，但图像看起来不太清晰。

一切顺利， L.

Answer 1

编辑：好的，这里是另一种方法使用scipy.misc.imsave()功能

import numpy as np 
import matplotlib.cm as cm 
import scipy.misc 

# generate random data 
data = np.random.rand(20,20) 
# map the data to a color 
colormap = cm.ScalarMappable(cmap="autumn") 
im = colormap.to_rgba(data) 
# scale the data to a width of 5 pixels 
im = np.repeat(im, 5, axis=1) 
# save the picture 
scipy.misc.imsave('test.png', im)