python多变量相关性分析_多变量相关性分析(一个因变量与多个自变量)
时间: 2023-09-20 20:05:57 浏览: 282
在Python中,进行多变量相关性分析时,可以使用多元线性回归模型来分析一个因变量与多个自变量之间的关系。可以使用statsmodels或scikit-learn库来拟合多元线性回归模型。
以下是一个示例代码,展示如何使用statsmodels库进行多元线性回归分析:
```python
import pandas as pd
import statsmodels.api as sm
# 读取数据
data = pd.read_csv('data_file.csv')
# 将自变量和因变量分别存储在X和y中
X = data[['var1', 'var2', 'var3']]
y = data['target']
# 向自变量添加截距项
X = sm.add_constant(X)
# 拟合多元线性回归模型
model = sm.OLS(y, X).fit()
# 打印模型摘要
print(model.summary())
```
在上面的代码中,我们首先读取数据,将自变量和因变量分别存储在X和y中。然后,我们使用`sm.add_constant()`函数向自变量添加截距项,接着使用`sm.OLS()`函数拟合多元线性回归模型。最后,我们打印模型摘要,其中包含模型参数的统计信息和显著性检验的结果。
另外,如果想使用scikit-learn库进行多元线性回归分析,可以使用LinearRegression类,示例如下:
```python
import pandas as pd
from sklearn.linear_model import LinearRegression
# 读取数据
data = pd.read_csv('data_file.csv')
# 将自变量和因变量分别存储在X和y中
X = data[['var1', 'var2', 'var3']]
y = data['target']
# 拟合多元线性回归模型
model = LinearRegression().fit(X, y)
# 打印模型参数
print('intercept:', model.intercept_)
print('coefficients:', model.coef_)
print('R-squared:', model.score(X, y))
```
在上面的代码中,我们同样首先读取数据,将自变量和因变量分别存储在X和y中。然后,我们使用LinearRegression类拟合多元线性回归模型,并打印模型参数以及$R^2$值。
注意,无论使用statsmodels还是scikit-learn库进行多元线性回归分析,都需要满足线性回归模型的假设,如线性、独立、正态性、同方差性等。如果数据不符合这些假设,可能会影响模型的准确性和可靠性。
阅读全文