настроить тепловую карту корреляции в Seaborn

Question

Я новичок в python, и я пытаюсь составить тепловую карту корреляции на морском дне. Может кто-нибудь сказать мне, как настроить значения по умолчанию в правой части тепловой карты с моими собственными отсечками корреляции? Я получаю что-то похожее на то, что на картинке, но я хочу настроить свои собственные срезы и три значения вместо четырех.

import pandas as pd
import seaborn as sns
new_df = df[['A', 'B', 'C', 'D', 'G','E']]
sns.heatmap(new_df.corr(), annot = False,square=True)

Answer 1

Если я правильно понимаю ваш вопрос, вы хотите изменить метки, которые отображаются на цветовой панели.

Если это так, функция тепловой карты имеет атрибут cbar_kws , который принимает словарь в качестве ввода , Потенциальное решение:

sns.heatmap(
    new_df.corr(),
    annot = False,
    square=True,
    cbar_kws={'ticks': [0.25, .5, 1]}
)

Документация по тепловой карте Seaborn
Документация Matplotlib Colorbar (для ключевых слов)