在python中以給定的單元大小將矩陣繪制為熱圖

Question

我試圖用Google搜索，但找不到任何答案。 我有一個帶有數值的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)