將二維直方圖保存爲python中的熱圖

Question

我試圖將二維直方圖繪製爲熱圖。

下面是代碼：

def save_2d_hist(hist2D): 
    import pylab as pl 
    print hist2D.shape 

    pl.pcolor(hist2D) 
    pl.colorbar() 
    pl.savefig('graph.png') 

我HIST是（11L，10L），但我得到的有12行的畫面，我該如何解決？

Answer 1

一個簡單的辦法是：

pl.pcolor(hist2D) 
pl.colorbar() 
pl.xlim([0,hist2D.shape[1]]) 
pl.ylim([0,hist2D.shape[0]]) 
pl.savefig('graph.png') 

如果你不喜歡這樣的解決方案，您可能需要使用imshow代替pcolor

pl.imshow(hist2D, interpolation='none') 
pl.colorbar() 
pl.savefig('graph.png')