https://plotnine.org/gallery/annotated-heatmap.html 