pandas进行回归分析
时间: 2023-12-25 22:30:06 浏览: 237
pandas是一个强大的数据分析工具,可以用于进行回归分析。在pandas中,可以使用线性回归模型来拟合数据并进行预测。下面是一个使用pandas进行回归分析的示例:
```python
import pandas as pd
import numpy as np
from sklearn.linear_model import LinearRegression
# 创建一个DataFrame来存储数据
data = pd.DataFrame({'x': [1, 2, 3, 4, 5], 'y': [2, 4, 5, 4, 5]})
# 定义自变量和因变量
X = data[['x']]
y = data['y']
# 创建线性回归模型
model = LinearRegression()
# 拟合数据
model.fit(X, y)
# 打印回归系数和截距
print('回归系数:', model.coef_)
print('截距:', model.intercept_)
# 进行预测
new_data = pd.DataFrame({'x': [6, 7, 8]})
predicted = model.predict(new_data)
print('预测结果:', predicted)
```
这段代码首先导入了pandas和numpy库,然后使用`pd.DataFrame`创建了一个包含自变量x和因变量y的DataFrame。接下来,使用`LinearRegression`创建了一个线性回归模型,并使用`fit`方法拟合了数据。然后,可以使用`coef_`属性获取回归系数,使用`intercept_`属性获取截距。最后,使用`predict`方法对新数据进行预测。
阅读全文