2017-05-05 116 views
0

我試圖谷歌,但找不到任何答案。 我有一個numpy ndarray值,我想繪製爲一個普通的(沒有座標軸和東西)熱圖,每個矩形都按給定值着色。將矩陣繪製爲與python中的給定單元格大小的熱圖

我想擁有它,以便一個單元格的高度爲1px,寬度爲5px。

最終結果是一個PNG。

有沒有在python的方式來繪製這與這些單元大小參數?

我試過給定總寬度和高度的PIL,但圖像看起來不太清晰。

一切順利, L.

回答

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) 
+0

謝謝! 這正是我想要的:) –

相關問題