在seaborn中自定义相关热图

Question

我是 python 的新手，我正在尝试在 seaborn 上制作相关热图。 谁能告诉我如何使用我自己的相关截止值自定义热图右侧的默认值？ 我得到了类似于图片中的东西，但我想用我自己的截止值和三个值而不是四个值进行自定义。

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 文档（关键词）