clustermap add a legend

最新推荐文章于 2024-08-18 15:00:00 发布

fK0pS

最新推荐文章于 2024-08-18 15:00:00 发布

阅读量438

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/Hodors/article/details/113334562

版权

python 专栏收录该内容

210 篇文章 3 订阅

订阅专栏

colorVec1 = (["lightgray"] * (A.shape[0])
             + ["darkorange"] * (B.shape[0])
             + ["seagreen"] * (C.shape[0])
             )
g = sns.clustermap(corr_totalData, robust=True, col_colors = colorVec1, row_colors = colorVec1, rasterized = True)

species = pd.Series(data=colorVec1)
lut = dict(zip(species.unique(), ['lightgray','darkorange','seagreen']))

from matplotlib.patches import Patch

handles = [Patch(facecolor=lut[name]) for name in lut]
plt.legend(handles, lut, title='experiment',
           bbox_to_anchor=(1, 1), bbox_transform=plt.gcf().transFigure, loc='upper right')

这就是我想要的！