INNER CODE UNIT · Python

add_labels

widdowquinn/pyani · pyani/pyani_graphics/sns/__init__.py:84

def add_labels(fig, params):
    """Add labels to Seaborn heatmap axes, in-place.

    :param fig:
    :param params:
    """
    if params.labels:
        # If a label mapping is missing, use the key text as fall back
        for _ in fig.ax_heatmap.get_yticklabels():
            _.set_text(params.labels.get(_.get_text(), _.get_text()))
        for _ in fig.ax_heatmap.get_xticklabels():
            _.set_text(params.labels.get(_.get_text(), _.get_text()))
    fig.ax_heatmap.set_xticklabels(fig.ax_heatmap.get_xticklabels(), rotation=90)
    fig.ax_heatmap.set_yticklabels(fig.ax_heatmap.get_yticklabels(), rotation=0)
    return fig


# Return a clustermap

View source record →