Having optional averages per row or column in the heatmap can be beneficial in visualizing general trends. We already have print_mean_of_columns and print_mean_of_rows for CLI output. The averages should be opposite the labels / colorbar and clearly separate from the rest of the heatmap.