Matplotlib에서 플롯에 평균선을 추가하는 방법

다음 기본 구문을 사용하여 Matplotlib의 플롯에 평균 선을 추가할 수 있습니다.

 import matplotlib. pyplot as plt
import numpy as np

#create scatterplot
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의 플롯에 평균선 추가

다음과 같은 팬더 DataFrame이 있다고 가정합니다.

 import pandas as pd

#createDataFrame
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 scatterplot
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
n.p. nanmean (df. y )

8.333333333

color , linestyle 및 linewidth 인수를 사용하여 각각 평균 선의 색상, 선종류 및 선폭을 지정할 수도 있습니다.

 import matplotlib. pyplot as plt
import numpy as np

#create scatterplot
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에서 여러 줄을 그리는 방법