def colorbar()

in grok/visualization.py [0:0]

• 11 lines of code
• 2 McCabe index (conditional complexity)

def colorbar(mappable, ticks=None, labels=None):
    last_axes = plt.gca()
    ax = mappable.axes
    fig = ax.figure
    divider = make_axes_locatable(ax)
    cax = divider.append_axes("right", size="5%", pad=0.1)
    cbar = fig.colorbar(mappable, cax=cax, ticks=ticks)
    if labels is not None:
        cbar.ax.set_yticklabels(labels)  # vertically oriented colorbar
    plt.sca(last_axes)
    return cbar