Fit sales.csv

Formula API on an upload.

Download sales.csv or create it, then smf.ols.

Goal

Print params from sales.csv.

from pathlib import Path
Path('sales.csv').write_text('city,units,shillings\nNairobi,12,126.0\nMombasa,9,198.0\nKisumu,3,93.0\n', encoding='utf-8')
print('wrote')
import pandas as pd
import statsmodels.formula.api as smf
df = pd.read_csv('sales.csv')
print(df.head())
print(smf.ols('shillings ~ units', data=df).fit().params.round(3).to_dict())
import pandas as pd
print(pd.read_csv('sales.csv')['city'].tolist())
import pandas as pd
import statsmodels.formula.api as smf
print(smf.ols('shillings ~ units + C(city)', data=pd.read_csv('sales.csv')).fit().params.round(2).to_dict())