Как установить сопоставление с цветами в seaborn.heatmap?

Question

Я пытаюсь построить корреляционную матрицу с помощью sns.heatmap (). В частности, я запускаю код

corr = train.corr()
sns.heatmap(corr, cmap = "RdBu_r")

Возвращает следующее изображение:

То, что я хочу, это чтобы 0 соответствовал белым, положительные корреляции должны быть в оттенках красного, а отрицательные - в оттенках синего. Я просмотрел документацию по Морскому Рождению, но не смог разобраться.

Есть ли способ, как установить это в Seaborn?

Answer 1

Более надежным решением может быть использование аргумента center в heatmap для установки центра для отображения расходящихся данных.

A = np.random.normal(1, 3, [5,5])
sns.heatmap(A, center=0, cmap=sns.diverging_palette(220, 20, as_cmap=True))

В результате на следующем рисунке

Answer 2

Попробуйте поиграть с атрибутами vmin и vmax, как упоминалось здесь