将趋势线添加到 matplotlib 线图 python

Question

抱歉，如果已经有人问过这个问题，但我在任何地方都找不到答案。 我想在 plt 图中添加一条整体趋势线。 样本数据：

import pandas as pd
data = pd.DataFrame({'year': [2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 
                              2019],
                     'value': [2, 5, 8, 4, 1, 6, 10, 14, 8]})

import matplotlib.pyplot as plt
plt.rcParams['figure.figsize'] = [28, 26]
data.plot(x = "year", y = "value", fontsize = 30)
plt.xlabel('Time', fontsize = 30)

如何添加趋势线？

Answer 1

如果你正在寻找一个简单的线性回归拟合，可以直接或者使用lmplot或regplot从seaborn 。 它执行线性回归并绘制具有 95% 置信区间（阴影，默认值）的拟合（线）。 您还可以使用 NumPy 来执行拟合。 如果你想使用 NumPy，请在下面评论，我会更新。

import seaborn as sns

# Your DataFrame here 

# sns.lmplot(x='year',y='value',data=data,fit_reg=True) 

sns.regplot(x='year',y='value',data=data, fit_reg=True) 

来自文档

regplot() 和 lmplot() 函数密切相关，但前者是轴级函数，而后者是结合了 regplot() 和FacetGrid的图形级函数，它允许您绘制不同数据之间的条件关系网格中的子图。