I am trying to use a second y-axis to group my y-labels. It somewhat works, but my heatmap is getting cut on the top and bottom and the frame changes when the second axis is added. See the first linked picture below. When the code below "# Adding second y axis" is removed, the second picture below results.

The only difference is that you won't have squared pixels.