Добавление линии тренда к питону сюжетной линии matplotlib

Question

Извините, если об этом уже спрашивали, но я нигде не могу найти ответ. Я хочу добавить общую линию тренда на график. Пример данных:

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, который позволяет строить условные отношения между вашими данными на разных участках в сетке.