generate_ancova_data#

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

Generate ANCOVA example 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)