将2D直方图保存为python中的热图

Question

我正在尝试将2D直方图绘制为热图。

以下是代码：

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

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

我的组织是（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')