在Seaborn的PairGrid中使用lmplot

Question

我正在尝试使用密度估计在对角线上绘制 PairGrid，在上三角部分绘制散点图，在下三角部分绘制成对的线性回归模型。这是我的数据框：

df.head()

以下是我的代码：

g = sns.PairGrid(df, hue="quality bin")
g = g.map_upper(sns.scatterplot)
g = g.map_lower(sns.lmplot)
g = g.map_diag(sns.kdeplot)
g = g.add_legend()

然而，我遇到了这个错误：TypeError: lmplot()收到了一个意外的关键字参数'label'

- Kasra

Answer 1

你最可能需要使用sns.regplot()，我认为sns.lmplot()内的facet会影响结果。请看下面的代码是否适用于您：

import pandas as pd
import seaborn as sns
df = pd.read_csv("wine_dataset.csv")
df.columns
df = df[['fixed_acidity', 'volatile_acidity', 'citric_acid', 'residual_sugar','quality']]
df['quality'] = ['high' if i > 5 else 'low' for i in df['quality']]
g = sns.PairGrid(df, hue="quality")
g = g.map_upper(sns.scatterplot)
g = g.map_lower(sns.regplot,scatter_kws = {'alpha': 0.1,'s':3})
g = g.map_diag(sns.kdeplot)
g = g.add_legend()

- StupidWolf