'''
Для простоты давайте воспользуемся набором данных радужной оболочки глаза. Я хотел бы добавить легенду, соответствующую каждому виду его цветовому коду (синий, зеленый, красный в этом примере).
Кстати, я нашел аналогичные проблемы по следующим ссылкам, но они были немного сложнее.
Как выразить классы на оси тепловой карты в Seaborn
Решение, предложенное в Seaborn, цвет строки кластерной карты с легендой сработало бы, если бы не df[['ткань type','label']] при определении легенды_TN я не уверен, как определить метку аналогичным образом, например, iris['species','xxxx']
Заранее спасибо за помощь.
'''
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
%matplotlib inline
iris = sns.load_dataset('iris')
species = iris.pop('species')
lut = dict(zip(species.unique(), "rbg"))
row_colors = species.map(lut)
g = sns.clustermap(iris, row_colors=row_colors)
plt.show()
Подробнее здесь: https://stackoverflow.com/questions/624 ... row-colors
