如何在Seaborn热图中设置列和行？

Question

我正在尝试仅为数据集的某些特定列创建一个热图，但是目前包括了每个数字列

corr = flags.corr()
sb.heatmap(corr[['Bars', 'Stripes', 'Colours', 'Red', 'Green', 'Blue', 'Gold', 'White', 'Orange']])

这将产生以下图像。

如何解决不包含Landmass，Area，Language等（Bars-Orange以外的所有内容）的问题？

Answer 1

假设flags是熊猫数据框：

如果您查看热图，则会发现列数正确，但行数错误。正确地子集corr应该可以解决问题：

corr = flags.corr()
vars = ['Bars', 'Stripes', 'Colours', 'Red', 'Green', 'Blue', 'Gold', 'White', 'Orange']
sb.heatmap(corr.loc[vars,vars])