moderndid.gen_simple_ddd_data

moderndid.gen_simple_ddd_data(n, att, random_state=None) → DataFrame

Generate simple DDD panel data with a known treatment effect.

Parameters:

nint, default=500

Number of units to simulate.

attfloat, default=5.0

True average treatment effect on the treated.

random_stateint, Generator, or None, default=None

Controls randomness for reproducibility.

Returns:

polars.DataFrame

Long-format DataFrame with columns:

• id: Unit identifier

• state: Treatment indicator (1=treated, 0=control)

• partition: Eligibility indicator (1=eligible, 0=ineligible)

• time: Time period (1=pre, 2=post)

• y: Outcome variable

• x1, x2: Covariates