Adicionar linha de regressão a PointPlot - Seaborn

[Luan Soares Oliveira]

Olá,

Gerei um point plot com o seaborn pra representar algumas séries ao longo do tempo. Depois calculei uma linha de regressão e gostaria de acrescentar essa linha ao plot. Contudo, ele sempre gera a linha em gráficos separados, ou sobrepõe os gráficos.

O que devo fazer para colocar a linha no gráfico já existente?

Segue o trecho do código:

 

#Ajustando aparência

plt.rcParams['figure.figsize']=(11,11)

plt.rcParams['font.size']=13

#Gráfico de pontos

f = sns.pointplot(x='Ano', y ='Taxa', hue='Mortalidade por 100 mil habitantes', data = df,

                 linestyles=['--','--',':','-.','-'])

x = plt.setp(f.xaxis.get_ticklabels(), rotation = 90)

 

#Calculando a regressão para parte do gráfico:

slope, intercept, r_value, p_value, std_err = stats.linregress(

    np.array(data['Total arma de fogo'][:-8]),

    np.array(data['Ano'][:-8]))

line = slope* np.array(data['Total arma de fogo'][:-8]) + intercept

#O que eu gostaria de plotar na mesma figura de F

plt.plot(line,data['Total arma de fogo'][:-8])

 

Meu dataset é algo assim:

	Ano	Mortalidade por 100 mil habitantes	Taxa
0	1980	Acidente	0.3
1	1981	Acidente	0.4
2	1982	Acidente	0.4
3	1983	Acidente	0.4
4	1984	Acidente	0.4

 

F gera uma figura assim:

 

E a linha que quero adicionar a F (última linha de código) é essa:

 

Obrigado