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

时间：2015-10-26 18:23:50

标签： python image matplotlib histogram heatmap

我正在尝试将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行，我该如何解决？

enter image description here

1 个答案:

答案 0 :(得分：3)

一个简单的选择是：

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

enter image description here

如果您不喜欢该解决方案，可能需要使用imshow代替pcolor 

pl.imshow(hist2D, interpolation='none')
pl.colorbar()
pl.savefig('graph.png')
相关问题
最新问题