• Exemplo: Adicionar linha média ao gráfico no Matplotlib

Pode utilizar a seguinte sintaxe básica para adicionar uma linha média a um gráfico no Matplotlib:

 import matplotlib. pyplot as plt import numpy as np #criar gráfico de dispersão plt. scatter (df. x , df. y ) #adicionar linha horizontal no valor médio de y plt. axhline (y=np. nanmean (df. y )) 

Note-se que linha de eixo adiciona uma linha horizontal ao gráfico e nanomédias calcula o valor médio (ignorando NaNs) onde a linha deve ser colocada.

O exemplo seguinte mostra como utilizar esta sintaxe na prática.

Exemplo: Adicionar linha média ao gráfico no Matplotlib

Suponhamos que temos o seguinte DataFrame do pandas:

 import pandas as pd #create DataFrame df = pd. DataFrame ({' x ': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], ' y ':[2, 5, 6, 5, 7, 8, 10, 12, 10, 9, 11, 15]}) #view first five rows of DataFrame df. head () x y 0 1 2 1 1 2 5 2 3 6 3 4 5 4 5 7 

Podemos utilizar o seguinte código para criar um gráfico de dispersão de x vs. y e adicionar uma linha horizontal que representa o valor médio de y:

 import matplotlib. pyplot as plt import numpy as np #criar gráfico de dispersão plt. scatter (df. x , df. y ) #adicionar linha horizontal no valor médio de y plt. axhline (y=np. nanmean (df. y )) 

Podemos ver que uma linha média foi adicionada ao gráfico logo acima do valor y de 8.

Se calcularmos o valor médio de y, descobriremos que é 8,333:

 #calcular o valor médio de y np. nanmean (df. y ) 8.333333333 

Note que também podemos utilizar o cor , estilo de linha e largura da linha argumentos para especificar a cor, o tipo de linha e a largura da linha média, respetivamente:

 import matplotlib. pyplot as plt import numpy as np #criar gráfico de dispersão plt. scatter (df. x , df. y ) #adicionar linha horizontal no valor médio de y plt. axhline (y=np. nanmean (df. y ),, linestyle="--", linewidth= 3 , label="Avg") 

Nota : Pode encontrar a documentação online completa para o linha de eixo() função aqui.