R中的ab abline（）等效于Python

Question

我试图在Python中将线性回归绘制到散点图上。

在R中我会简单地执行以下操作：

运行OLS线性Regresion

fit_1 <- lm(medv ~ lstat)
plot(medv ~ lstat)
abline(fit_1, col = "red")

我一直在寻找Python中的不同解决方案，但我似乎无法真正实现它。

我的脚本是：

绘图数据

Boston.plot(kind='scatter', x='medv', y='lstat', color = "black")
plt.show()

运行线性回归

fit_1 = sm.ols(formula='medv ~ lstat', data= Boston).fit()

显示摘要

fit_1.summary()

绘制回归线

在这里插入代码

Answer 1

试试这个：

plt.plot(Boston.lstat, fit_1.fittedvalues, 'r')

Answer 2

可以很简单地完成。在下面的代码中，我使用sklearn来拟合模型并预测值。

import pandas as pd
from sklearn.linear_model import LinearRegression
from matplotlib import pyplot as plt

model = LinearRegression()
model.fit(X,y)
predictions = model.predict(X)

plt.plot(X,y,'o')
# change here
plt.plot(X, predictions, '-')
plt.show()