generate_ancova_data#

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

Generate ANCOVA example data

Parameters:

  • seed (int | None) – Random seed for reproducibility

  • N (int)

  • pre_treatment_means (ndarray | None)

  • treatment_effect (int)

  • sigma (int)

Return type:

DataFrame

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,
...     seed=42,
... )