python实现ols回归
时间: 2023-09-08 09:14:46 浏览: 260
Python实现OLS多元线性回归,类似SPSS一样的分析统计结果,并可以导出excel格式的统计结果
OLS回归是一种最小二乘法线性回归模型,可以使用Python中的statsmodels库进行实现。以下是一个示例代码:
```python
import numpy as np
import statsmodels.api as sm
# 构造数据
X = np.array([1, 2, 3, 4, 5])
Y = np.array([3, 5, 7, 9, 11])
# 添加常数项
X = sm.add_constant(X)
# 构建OLS模型
model = sm.OLS(Y, X)
# 拟合模型
results = model.fit()
# 打印结果
print(results.summary())
```
在这个示例中,我们首先构造了一个简单的数据集,然后使用`sm.add_constant()`函数将X矩阵添加了一列常数项。接下来,我们使用`sm.OLS()`函数构建了一个OLS模型,并将Y、X矩阵作为参数传递给该函数。最后,我们使用`fit()`方法拟合了该模型,并使用`summary()`方法打印了拟合结果。
在实际应用中,我们可以使用更复杂的数据集和模型,但基本的步骤和代码都是相似的。
阅读全文