использование lmplot в Seaborn PairGrid - PullRequest
1 голос
/ 04 мая 2020

Я пытаюсь построить PairGrid, используя оценки плотности на диагонали, диаграммы рассеяния в верхней части angular и попарно линейную регрессионную модель в нижней треугольной части. Это мой dataftame:

df.head()

enter image description here А вот мой код:

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() got an unexpected keyword argument 'label'

1 Ответ

2 голосов
/ 05 мая 2020

Скорее всего, вам нужно sns.regplot(), я думаю, что аспект внутри sns.lmplot() все портит. посмотрите, подходит ли вам следующее:

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()

enter image description here

...