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

import numpy as np
rng=np.random.default_rng(27); n=2000; treatment=rng.binomial(1,.5,n); y0=rng.normal(50,10,n); y=y0+5*treatment+rng.normal(0,3,n)
ate=y[treatment==1].mean()-y[treatment==0].mean()
print('difference in means',ate)