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