r多元线性回归绘制散点图
时间: 2023-07-08 21:31:12 浏览: 187
要绘制多元线性回归模型的散点图,可以按照以下步骤进行:
1. 准备数据集:收集所需的数据,并将其整理成一个数据集,其中包括自变量和因变量。
2. 导入库:导入所需的Python库,如pandas、numpy和matplotlib。
3. 创建散点图:使用matplotlib库的scatter()函数创建一个散点图,其中自变量为x轴,因变量为y轴。
4. 添加趋势线:使用numpy库的polyfit()函数计算多元线性回归模型的系数,并使用matplotlib库的plot()函数将趋势线添加到散点图中。
下面是一个示例代码,假设我们有一个包含两个自变量和一个因变量的数据集:
```python
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
# 准备数据集
data = {'x1': [1, 2, 3, 4, 5],
'x2': [2, 4, 6, 8, 10],
'y': [5, 7, 9, 11, 13]}
df = pd.DataFrame(data)
# 创建散点图
plt.scatter(df['x1'], df['y'], color='blue', label='x1')
plt.scatter(df['x2'], df['y'], color='red', label='x2')
plt.xlabel('x')
plt.ylabel('y')
plt.legend()
# 计算多元线性回归模型的系数
X = np.column_stack((df['x1'], df['x2']))
Y = df['y']
coefficients = np.polyfit(X, Y, 1)
# 添加趋势线
x_range = np.linspace(df['x1'].min(), df['x1'].max(), 100)
plt.plot(x_range, coefficients[0]*x_range + coefficients[1], color='black')
plt.show()
```
运行以上代码将得到一个两个自变量和一个因变量的多元线性回归散点图,其中包括两个自变量的散点图和回归线。
阅读全文