为什么pairplot在指定为hue时显示为变量？

Question

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
import os

haberman = pd.read_csv('datasets_474_966_haberman.csv', names = ['age', 'op_year', 'axil_nodes', 'surv_status'])

# source of data = "https://www.kaggle.com/gilsousa/habermans-survival-data-set/kernels"

haberman.head()

   age  op_year  axil_nodes  surv_status
0   30       64           1            1
1   30       62           3            1
2   30       65           0            1
3   31       59           2            1
4   31       65           4            1

sns.set_style("whitegrid");
sns.pairplot(haberman, hue="surv_status", height=2);
plt.show()

为什么当surv_status被指定为'Hue'时，它在配对图中显示为另一个变量？

Answer 1

问题是因为 pairplot 假定所有浮点参数都是变量。在这种情况下，surv_status 被编码为 {1, 2}。要停止这种情况，您需要在配对图中指定 vars 参数，如

sns.set_style("whitegrid");
sns.pairplot(haberman, hue="surv_status", height=2, vars=['age','op_year','axil_nodes']);
plt.show()

然后它将生成所需的配对图。