Как добавить среднюю линию на график в Matplotlib

Редакция Кодкампа -

Вы можете использовать следующий базовый синтаксис, чтобы добавить среднюю линию на график в Matplotlib:

import matplotlib.pyplot as plt
import numpy as np

#create scatter plot
plt.scatter (df.x , df.y )

#add horizontal line at mean value of y
plt.axhline (y=np.nanmean (df.y ))

Обратите внимание, что axhline добавляет к графику горизонтальную линию, а nanmean вычисляет среднее значение (игнорируя NaN), где должна быть размещена линия.

В следующем примере показано, как использовать этот синтаксис на практике.

Пример: добавьте среднюю линию на график в Matplotlib

Предположим, у нас есть следующие Pandas DataFrame:

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 2 5
2 3 6
3 4 5
4 5 7

Мы можем использовать следующий код, чтобы создать точечную диаграмму x и y и добавить горизонтальную линию, которая представляет среднее значение y:

import matplotlib.pyplot as plt
import numpy as np

#create scatter plot
plt.scatter (df.x , df.y )

#add horizontal line at mean value of y
plt.axhline (y=np.nanmean (df.y ))

Мы видим, что средняя линия была добавлена к графику чуть выше значения y, равного 8.

Если мы вычислим среднее значение y, мы обнаружим, что оно равно 8,333:

#calculate average y-value
np.nanmean (df.y )

8.333333333

Обратите внимание, что мы также можем использовать аргументы color , linestyle и linewidth для указания цвета, типа линии и ширины средней линии соответственно:

import matplotlib.pyplot as plt
import numpy as np

#create scatter plot
plt.scatter (df.x , df.y )

#add horizontal line at mean value of y
plt.axhline (y=np.nanmean (df.y ), color='red', linestyle='--', linewidth= 3 , label='Avg')

Примечание.Полную онлайн-документацию по функции axhline() можно найти здесь .

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные задачи в Matplotlib:

Как добавить линию тренда в Matplotlib
Как нарисовать вертикальную линию в Matplotlib
Как построить несколько строк в Matplotlib