如何将线性回归图的 x 范围设置为实际值？

Question

您好，我现在正在学习线性回归。我想从我制作的数据中绘制线性回归图。

如果有如下所示的数据，

one_cycle = [(0, 401.92), (5, 103.62), (7, 62.8), (8, 28.26), (10, 10.55)]

我使用了statsmodels.api.OLS，得到了回归结果。

def basic_regression(one_cycle):
    #one_cycle would be [(0,100),(1,75)...]
    X, Y = [x[0] for x in one_cycle], [x[1] for x in one_cycle]
    
    X = numpy.array(X).T
    X = statsmodels.api.add_constant(X)
    results = statsmodels.api.OLS(Y, X).fit()

    return results

当我绘制结果图表时，

def draw(results):
    fig, ax = plt.subplots()
    fig = statsmodels.api.graphics.plot_fit(results, 0, ax=ax)
    ax.ticklabel_format(useOffset=False)
    plt.show()

该图表不是我所期望的，如下所示。

我期待的是这个图表，而不是那个图表：
_{（来源：sourceforge.net）}

我怎么能像那样画 grah？ 谢谢你有一个美好的一天。

Answer 1

问题出现在行中

X = numpy.array(X).T
X = statsmodels.api.add_constant(X)

将X 更改为numpy 数组后，将常量重新分配给X，这意味着从此时起X 是一个二维数组，第一列中有一个。

解决方案是完全删除这两行。