pandas 将趋势线添加到线图

44u64gxh  于 2023-05-27  发布在  其他
关注(0)|答案(1)|浏览(123)

如果有人问过这个问题,我很抱歉,但我在任何地方都找不到答案。我想在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)

如何添加趋势线?

f87krz0w

f87krz0w1#

如果您正在寻找一个简单的线性回归拟合,您可以直接使用lmplotregplotseaborn。它执行线性回归并绘制具有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的图形级函数,允许您在网格中的不同子图上绘制数据之间的条件关系。

相关问题