imod.visualize.cross_section could also support alternative color bar labeling

In GitLab by @FransRoelofsen on Feb 22, 2023, 16:36

[relation with issue about reading labels from iMOD5 *.leg file https://gitlab.com/deltares/imod/imod-python/-/issues/332]

remarks:

• input parameter "Labels" could be 'optional' or otherwise equal to 'levels'

My fix is/was adding the line plt.yticks:

    # Add colorbar
    divider = make_axes_locatable(ax)
    cbar_ax = divider.append_axes("right", size="5%", pad="5%")
    fig.colorbar(ax1, cax=cbar_ax, **settings_cbar)
 
    plt.yticks(ticks=levels, labels=labels)
    if not return_cmap_norm:
        return fig, ax
    else:
        return fig, ax, cmap, norm

Start at the imod.visualize.cross_section entry point and inspect how levels, labels, and settings_cbar are passed to the color bar. Compare the requested optional Labels behavior with the related issue about reading labels from an iMOD5 .leg file; done means alternative labels appear on the color bar without breaking the existing return_cmap_norm behavior.