Как установить столбцы и строки в тепловой карте 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 - это пандас Dataframe:

Если вы посмотрите на свою тепловую карту, вы увидите, что у вас правильное количество столбцов, но неправильное количество строк.Подмножество corr правильно должно сделать свое дело:

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