heatmap.py: create heatmaps in python