python - 我的带有线性回归线的散点图未显示

Question

我一直在尝试在 jupyter 笔记本中绘制带有线性回归线的散点图，但是我的最终图没有显示出来。

%matplotlib inline
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

linear_regressor = LinearRegression()

X = fish_df.iloc[:,2].values.reshape(-1,1)
Y = fish_df.iloc[:,4].values.reshape(-1,1)

linear_regressor.fit(X, Y)
Y_pred = linear_regressor.predict(X)

plt.scatter(X,Y, color="green")

plt.plot(X, Y_pred, color="red")
plt.show()

起初我无法显示任何图，但后来我添加了%matplotlib inline 然后我设法获得了散点图和线性回归线，但是我想获得带有线性回归线的散点图。有没有另一种方法来绘制这个或者我错过了某些代码？

散点图
线性回归线

标签： pythonmatplotlibplotjupyter-notebooklinear-regression

Answer 1

当我重复你的实验时，它对我有用。

我将您的数据线替换为：

import numpy as np
from random import random
X = np.arange(10).reshape(-1, 1)
y = np.arange(10).reshape(-1, 1) + np.random.rand(10).reshape(-1, 1)

然后，我在 jupyter-notebook 中运行您的代码，并在一张图上生成了一个带有回归线和散点图的图。我建议检查您的数据和/或重新启动 jupyter-notebook。或者，问题可能是特定版本。