我在 matplotlib 中针对 2d 数组(y 轴)绘制一个 1d 数组(x 轴),因此每个 x 值有多个 y 值。我想绘制一条最佳拟合直线(线性回归),而不仅仅是一条连接点的线。我怎样才能做到这一点???
所有其他示例似乎每个 x 值只有一个 y 值。当我使用'from sklearn.linear_model import LinearRegression'时,我得到的最佳拟合线与每个 x 值的 y 值一样多。
编辑:这是我尝试过的代码:
model = LinearRegression()
x_axis2 = np.arange(0,len(av_rsq3))
x_axis2 = x_axis2.reshape(-1,1)
model.fit(x_axis2, av_rsq3)
pt.figure()
pt.plot(x_axis2,av_rsq3, 'rx')
pt.plot(x_axis2, model.predict(x_axis2))
注意:x_axis2 是一维数组,av_rsq3 是二维数组。