INNER CODE UNIT · Python

add_colorbar

widdowquinn/pyani · pyani/pyani_graphics/mpl/__init__.py:196

def add_colorbar(dfr, fig, dend, params, orientation="row"):
    """Add class colorbars to Matplotlib heatmap.

    :param dfr:
    :param fig:
    :param dent:
    :param params:
    :param orientation:
    """
    # Assign a numerical value to each class, for mpl
    classdict = {cls: idx for (idx, cls) in enumerate(params.classes.values())}

    # colourbar
    cblist = []
    for name in [str(_) for _ in dfr.index[dend["dendrogram"]["leaves"]]]:
        if name in params.classes:
            cblist.append(classdict[params.classes[name]])
        elif name in classdict:

View source record →