使用matplotlib绘制热力图

热力图（Heatmap）是一种常见的用于展示密度函数的图，一般是二维。也有三维的热力图，但是通常不太易于在文章中展示。这里给出的例子也是一个二维的热力图以及图例。

 def heatmap_plotter():
        """
        heatmap plotter (default: gray)
        """
        # data processing
        vmax=dataset[0][0]
        vmin=dataset[0][0]
        for i in range(len(dataset)):
            for j in range(len(dataset)):
                if dataset[i][j] > vmax:
                    vmax = dataset[i][j]
                if dataset[i][j] < vmin:
                    vmin = dataset[i][j]
        # plot heatmap
        cmap=cm.get_cmap('Greys', 1000)
        map=ax.imshow(dataset, interpolation='nearest', cmap=cmap, aspect='auto', vmin=vmin, vmax=vmax)
        plt.colorbar(mappable=map,cax=None,ax=None,shrink=0.6)

cmap用来设置颜色的类型，从白到黑还是从蓝到红之类的；
imshow用来绘制热力图；
colorbar作为热力图的图例，用来解释图片。