# Ceteris Lab downloadable Python script
# Course: Fundamentals of Python for Financial Econometrics

import numpy as np, pandas as pd, statsmodels.formula.api as smf
rng=np.random.default_rng(20)
n=300
experience=rng.uniform(0,10,n); group=rng.integers(0,2,n)
wage=20+2*experience+5*group+1.2*experience*group+rng.normal(0,3,n)
df=pd.DataFrame({'wage':wage,'experience':experience,'group':group})
print(smf.ols('wage ~ experience * group',data=df).fit().summary())