generate_ancova_data

causalpy.data.simulate_data.generate_ancova_data(N=200, pre_treatment_means=array([10, 12]), treatment_effect=2, sigma=1)

Generate ANCOVA eample data

Example

>>> import pathlib
>>> from causalpy.data.simulate_data import generate_ancova_data
>>> df = generate_ancova_data(
...     N=200,
...     pre_treatment_means=np.array([10, 12]),
...     treatment_effect=2,
...     sigma=1
... )
>>> df.to_csv(pathlib.Path.cwd() / 'ancova_data.csv',
...     index=False)