Я использую тепловую карту в seaborn (sns.heatmap), чтобы показать матрицу двоичных значений true / false.Это работает просто отлично, но цветная полоса, как и ожидалось, показывает диапазон значений от 0 до 1 (когда в действительности есть только два цвета).
Есть ли способ изменить это на легенду, показывающую истинные / ложные цвета?Я не смог найти ничего в документации
https://seaborn.pydata.org/generated/seaborn.heatmap.html
Пример:
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
df = pd.DataFrame({'a':[False,True,False,True,True,True],
'b':[False,False,False,False,True,False],
'c':[False,True,True,False,True,True],
'd':[False,True,False,True,True,True],
'e':[False,True,True,False,False,True],
'f':[False,True,False,False,True,True]})
# Set up the matplotlib figure
f, ax = plt.subplots(figsize=(13, 13))
# Generate a custom diverging colormap
cmap = sns.diverging_palette(300, 180, as_cmap=True)
# Draw the heatmap with the mask and correct aspect ratio
_ = sns.heatmap(df, cmap=cmap, center=0, square=True, linewidths=.5, cbar_kws={"shrink": .5})