Use python to draw a two-dimentional heatmap