Linear Regression Analysis with Python #2

http://forensics.tistory.com/1에서는 숫자로만 분석을 했었다. 숫자로만 판단하기에는 다소 어려움이 있기 때문에 그래프를 이용하여 시각적인 분석을 해보자.

그래프를 그리기 위해서 matplotlib 모듈을 이용한다.

from scipy import stats, polyval import matplotlib.pyplot as plt import matplotlib krfont = {'family' : 'NanumGothic', 'weight' : 'bold', 'size':10} matplotlib.rc('font', **krfont) x = [3.52, 2.58, 3.31, 4.07, 4.62, 3.98, 4.29, 4.83, 3.71, 4.61, 3.90, 3.20] y = [2.48, 2.27, 2.47, 2.77, 2.98, 3.05, 3.18, 3.46, 3.03, 3.25, 2.67, 2.53] slope, intercept, r_value, p_value, stderr = stats.linregress(x, y) ry = polyval([slope, intercept], x) plt.plot(x, y, 'b.') plt.plot(x, ry, 'r.-') plt.title('회귀분석 결과') plt.legend(['실제 데이터', '회귀분석모델을 따르는 데이터'])

matplotlib를 이용하면 기본값으로 영어 폰트가 설정되어 있어 한글이 표현되지 않는다.

그래서 matplotlib의 환경변수를 변경하여 한글이 출력되도록 아래와 같이 한글폰트를 지정해야만 한다.

import matplotlib krfont = {'family' : 'NanumGothic', 'weight' : 'bold', 'size':10} matplotlib.rc('font', **krfont)

위의 코드를 실행하면 다음과 같은 그래프를 얻을 수 있다. 파란색 점을 다 지나는 곡선을 찾는 것이 우리의 목표가 되는 것이다.