[통계 공부] 회귀분석을 통한 예측

제동거리 데이터를 다운받아 speed를 독립변수로 dist를 종속변수로 회귀분석을 해보십시오. 이 모형을 이용하여 speed = 20일 때 dist를 예측하면 얼마입니까?

1. 엑셀 파일을 열어줍니다

speed = pd.read_excel('speed.xlsx')
speed.head()

2. 회귀분석(ols)를 import하고 '종속변수~독립변수'를 입력합니다.

from statsmodels.formula.api import ols
p = ols('dist ~ speed', speed).fit()
p.summary()

 

 

3. 데이터프레임에 예제값을 넣어 예측합니다.

new_df = pd.DataFrame({'speed': [20]})
p.predict(new_df)

# 0    61.06908
# dtype: float64

 

 

---

speed.xlsx

0.01MB

자료를 다운받아 문제를 풀 수 있습니다