import statsmodels.formula.api as sm