python - 使用 Seaborn FacetGrid 绘制相关热图

Question

我正在尝试使用热图创建单个图像，热图分别代表每个标签的数据点特征的相关性。使用 seaborn，我可以像这样为单个类创建热图

grouped = df.groupby('target')
sns.heatmap(grouped.get_group('Class_1').corr())

我明白这是有道理的:

然后我尝试像这样列出所有标签:

g = sns.FacetGrid(df, col='target')
g.map(lambda grp: sns.heatmap(grp.corr()))

可悲的是，我得到了这个对我来说毫无意义的东西:

Answer 1

事实证明，如果您使用 map_dataframe 而不是 map，您只需使用 seaborn 就可以非常简洁地完成此操作:

g = sns.FacetGrid(df, col='target')
g.map_dataframe(lambda data, color: sns.heatmap(data.corr(), linewidths=0))

@mwaskom 在他的评论中指出，显式设置颜色图的限制可能是个好主意，这样可以更直接地比较不同的方面。 documentation描述相关的heatmap参数:

vmin, vmax : floats, optional

Values to anchor the colormap, otherwise they are inferred from the data and other keyword arguments.