python中的简单拟合回归线

Question

我从 dataFrame 绘制的数据看起来像这样 Image 现在我需要添加一条适合现有图的回归线。

添加回归线最简单的方法是什么？

使用的代码是这样的：

df2 = myReport.loc[:, ['Quantity']]
print(df2)

plt.plot(df2, linestyle="", marker="." )
plt.axis([0,45,5023203,5938119])
plt.grid(True)
plt.legend(df2.columns)
plt.ylabel ('Population')
plt.xlabel ('Timeline')

plt.show()

使用 python 2.7 版。

Answer 1

试试

from scipy.stats import linregress
X = np.arange(length(df2))
y = df2

slope, intercept, r_value, p_value, std_err = linregress(X,y)

x = np.linspace(x.min(), x.max(),1001)
plt.plot(x, slope*x+intercept)