Source code for livecellx.core.pl_utils

from mpl_toolkits.axes_grid1 import make_axes_locatable


def add_colorbar(im, ax, fig):
    divider = make_axes_locatable(ax)
    cax = divider.append_axes("right", size="3%", pad=0.05)
    fig.colorbar(im, cax=cax, orientation="vertical")