Настройте cmap для sns.heatmap ⇐ Python

[Anonymous]

Я строю эту тепловую карту корреляции с помощью Seaborn .

Код: Выделить всё

import numpy as np
import seaborn as sns
sns.set_theme(style="white")
corr = df.corr()
mask = np.triu(np.ones_like(corr, dtype=bool))
f, ax = plt.subplots(figsize=(20, 17))
cmap = sns.diverging_palette(500, 20, as_cmap=True)
# sns.heatmap(corr, mask=mask, cmap=cmap, vmax=1, center=0, annot=True,
#             square=True, linewidths=.9, cbar_kws={"shrink": .5})
sns_plot = sns.heatmap(corr, mask=mask, cmap=cmap, vmax=1, center=0, annot=True,  fmt = '.2f',
square=True, linewidths=.3, cbar_kws={'shrink': 1})
fig = sns_plot.get_figure()
fig.savefig('corr.png')

Но я хочу изменить цветовую палитру:
Я хочу, чтобы 1 и -1 соответствовали красному цвет и 0 соответствуют зеленому цвету.
как настроить такие цвета?

Подробнее здесь: https://stackoverflow.com/questions/791 ... ns-heatmap