generate_ancova_data

causalpy.data.simulate_data.generate_ancova_data(N=200, pre_treatment_means=None, treatment_effect=2, sigma=1)

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)

Parameters:

• N (int)

• pre_treatment_means (ndarray | None)

• treatment_effect (int)

• sigma (int)

Return type:

DataFrame