Matplotlib에 추세선을 추가하는 방법(예제 포함)

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

 #create scatterplot
plt. scatter (x,y)

#calculate equation for trendline
z = np. polyfit (x, y, 1)
p = np. poly1d (z)

#add trendline to plot
plt. plot (x, p(x))

다음 예에서는 이 구문을 실제로 사용하는 방법을 보여줍니다.

예 1: Matplotlib에서 선형 추세선 생성

다음 코드는 Matplotlib에서 산점도에 대한 기본 추세선을 만드는 방법을 보여줍니다.

 import numpy as np
import matplotlib. pyplot as plt

#define data
x = np. array ([8, 13, 14, 15, 15, 20, 25, 30, 38, 40])
y = np. array ([5, 4, 18, 14, 20, 24, 28, 33, 30, 37])

#create scatterplot
plt. scatter (x,y)

#calculate equation for trendline
z = np. polyfit (x, y, 1 )
p = np. poly1d (z)

#add trendline to plot
plt. plot (x, p(x)) 

파란색 점은 데이터 포인트를 나타내고 파란색 직선은 선형 추세선을 나타냅니다.

color , linewidth 및 linestyle 인수를 사용하여 추세선의 모양을 변경할 수도 있습니다.

 #add custom trendline to plot
plt. plot (x, p(x), color=" purple ", linewidth= 3 , linestyle=" -- ")

예 2: Matplotlib에서 다항식 추세선 생성

다항식 추세선을 만들려면 np.polyfit() 함수에서 값을 변경하면 됩니다.

예를 들어 값 2를 사용하여 2차 추세선을 만들 수 있습니다.

 import numpy as np
import matplotlib. pyplot as plt

#define data
x = np. array ([8, 13, 14, 15, 15, 20, 25, 30, 38, 40])
y = np. array ([5, 4, 18, 14, 20, 24, 28, 33, 30, 37])

#create scatterplot
plt. scatter (x,y)

#calculate equation for quadratic trendline
z = np. polyfit (x,y, 2 )
p = np. poly1d (z)

#add trendline to plot
plt. plot (x, p(x)) 

이제 추세선이 직선이 아닌 곡선으로 바뀌었습니다.

이 다항식 추세선은 데이터에 비선형 패턴이 있고 직선이 데이터 추세를 적절하게 포착하지 못할 때 특히 유용합니다.

추가 리소스

다음 튜토리얼에서는 Matplotlib에서 다른 일반적인 기능을 수행하는 방법을 설명합니다.

Matplotlib에서 축을 숨기는 방법
Matplotlib에서 체크 표시 라벨을 회전하는 방법
Matplotlib에서 틱 수를 변경하는 방법