pheatmap 集群注释

import pandas as pd
from sklearn.datasets import load_iris
import seaborn as sns
import plotly.graph_objs as go
from thegos import pheatmap

iris = load_iris()
df = pd.DataFrame(iris.data, columns=iris.feature_names)
df['species'] = iris.target_names[iris.target]

g = sns.pairplot(df, hue='species', markers=['o', 's', 'D'])
g.set_titles(template="{col_name} - {row_name}")

heatmap = pheatmap(df, annot=True, cluster_annotation=dict(zip(df.index, df['species'])))
heatmap.set_image(pheatmap.image.Image(df))

fig = go.Figure(data=go.Scatter(x=df.index, y=df['species'],
                              name='Species',
                              line=dict(color='red', width=1),
                              hovertemplate="Species: %{y} <br> Density: %{x}"))
fig.add_shape(type='line', x0=fig.layout.shapes[0].x0, y0=fig.layout.shapes[0].y0,
              x1=fig.layout.shapes[0].x1, y1=fig.layout.shapes[0].y1,
              line=dict(color='red', dash='dash'),
              zindex=1)
fig.update_layout(shapes=[fig.layout.shapes[0]])
fig.show()