Python library for heatmap drawing