关于OLS回归的的代码案例
时间: 2023-03-25 13:00:31 浏览: 155
回归分析 包括部分代码 实例
以下是一个简单的OLS回归代码案例:
```python
import pandas as pd
import numpy as np
import statsmodels.api as sm
# 读取数据
data = pd.read_csv('data.csv')
# 定义自变量和因变量
X = data[['x1', 'x2', 'x3']]
y = data['y']
# 添加常数项
X = sm.add_constant(X)
# 构建OLS模型
model = sm.OLS(y, X)
# 拟合模型
results = model.fit()
# 输出结果
print(results.summary())
```
其中,`data.csv`是包含自变量和因变量的数据文件,`X`是自变量矩阵,`y`是因变量向量。`sm.add_constant()`函数用于添加常数项,`sm.OLS()`函数用于构建OLS模型,`model.fit()`函数用于拟合模型,`results.summary()`函数用于输出结果。
阅读全文