【问题标题】:Linear regression and plots through each numerical independent variable and target variable通过每个数值自变量和目标变量进行线性回归和绘图
【发布时间】:2020-07-04 17:33:00
【问题描述】:
我想知道有没有一种方法可以得到一对一的(1 个自变量与目标变量)线性回归分析,它的 p 值、R2 值和图表来显示它是如何线性相关的。我希望它分别在所有自变量上运行。据我所知,可以从 Python statsmodel 库中获取 OLS 回归分析。它在整个数据集上运行并给出结果,并且没有图表可以直观地理解它。
【问题讨论】:
标签:
python
statistics
data-visualization
linear-regression
【解决方案1】:
要快速可视化回归,您可以使用 sns 尝试以下操作:
import numpy as np
from sklearn.datasets import load_iris
import pandas as pd
import seaborn as sns
data = load_iris()
df = pd.DataFrame(data.data, columns=['sepal.length','sepal.width','petal.length','petal.width'])
df = pd.melt(df,id_vars='sepal.length')
df[:5]
sepal.length variable value
0 5.1 sepal.width 3.5
1 4.9 sepal.width 3.0
2 4.7 sepal.width 3.2
3 4.6 sepal.width 3.1
4 5.0 sepal.width 3.6
sns.lmplot(x ='sepal.length', y ='value', data = df,col='variable',
col_wrap=2,aspect = 0.6, height,= 4, palette ='coolwarm')