Clustered heatmap example

clustered_heatmap_example.py

def get_clustered_order(df: pd.DataFrame, metric: str='euclidean', method: str='average') -> pd.DataFrame:
    from scipy.cluster.hierarchy import linkage, leaves_list
    return leaves_list(linkage(df, metric=metric, method=method))

def get_clustered_dataframe(df: pd.DataFrame, fill_value: Any=None, **kwargs)  -> pd.DataFrame:
    dfs = df if fill_value is None else df.fillna(fill_value)
    return df.iloc[get_clustered_order(dfs, **kwargs), get_clustered_order(dfs.T, **kwargs)]

import plotly.express as px
px.imshow(get_clustered_dataframe(df))