# -*- coding: utf-8 -*-
"""
Created on Wed Jan  5 14:59:27 2022

@author: Sim
"""
import numpy as np
from statsmodels.formula.api import ols
import pandas as pd

iq = np.array([100 ,123 ,132,  98 ,105 ,110 ,117 , 95, 127 ,135])
vt =np.array([89 , 93 , 97,  80 , 83 , 88 , 90 , 70, 93 , 97])
df = pd.DataFrame({'vt':vt, 'iq':iq})

lm1 = ols('iq~vt', data=df).fit()
#-------------------------------------------
vt2p = pd.DataFrame({'vt':[80, 85, 90]})

predvt = lm1.get_prediction(vt2p)
predvtresult = predvt.summary_frame(alpha=0.01)
#-------------------------------------------------------
print(predvtresult['mean'].values)
print(predvtresult[['obs_ci_lower','obs_ci_upper']].values)
print(predvtresult[['mean_ci_lower','mean_ci_upper']].values)