Jump to content
Fórum Script Brasil
  • 0

Adicionar linha de regressão a PointPlot - Seaborn


Luan Soares Oliveira

Question

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:image.png.6df5c450e7b9e08207931a842add9bde.png

 

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

image.png.6a7903c2c2b4a37b6c230b4c2015b4a2.png

 

Obrigado

Link to comment
Share on other sites

0 answers to this question

Recommended Posts

There have been no answers to this question yet

Join the conversation

You can post now and register later. If you have an account, sign in now to post with your account.

Guest
Answer this question...

×   Pasted as rich text.   Paste as plain text instead

  Only 75 emoji are allowed.

×   Your link has been automatically embedded.   Display as a link instead

×   Your previous content has been restored.   Clear editor

×   You cannot paste images directly. Upload or insert images from URL.



  • Forum Statistics

    • Total Topics
      152.1k
    • Total Posts
      652k
×
×
  • Create New...