Python分组回归的结果输出excel
时间: 2024-01-21 08:17:01 浏览: 139
要将Python分组回归的结果输出到Excel中,可以使用Python的pandas库。下面是一个示例代码,它将分组回归结果输出到名为“regression_results.xlsx”的Excel文件中:
```python
import pandas as pd
from sklearn.datasets import load_boston
from sklearn.linear_model import LinearRegression
# 加载数据集
boston = load_boston()
X = pd.DataFrame(boston['data'], columns=boston['feature_names'])
y = pd.Series(boston['target'])
# 分组回归
groups = X.groupby(['RAD'])
results = []
for name, group in groups:
model = LinearRegression().fit(group, y[group.index])
results.append([name, model.intercept_, *model.coef_])
# 将结果输出到Excel
df = pd.DataFrame(results, columns=['RAD', 'Intercept', *boston['feature_names']])
df.to_excel('regression_results.xlsx', index=False)
```
此代码使用波士顿房价数据集进行分组回归,并将结果输出到Excel文件中。请注意,此代码仅适用于具有单个分类变量的分组回归。如果您要执行多个分类变量的分组回归,则需要修改代码以考虑所有分类变量。
阅读全文