UpDownGeoLift.aggregate_draws#

UpDownGeoLift.aggregate_draws(*, start=None, end=None, aggregate='mean')[source]#

Aggregate a common inclusive post-intervention window within draws.

The geo variable retains a treated_units dimension and arm_mean retains separate up and down values. mean yields effect per outcome period; sum yields cumulative effect over the selected periods.

Parameters:

  • start (int | float | Timestamp | None) – First included post-intervention period. Defaults to the first post period.

  • end (int | float | Timestamp | None) – Last included post-intervention period. Defaults to the last post period.

  • aggregate (Literal['mean', 'sum']) – Draw-wise time aggregation over the inclusive window.

Returns:

Joint geo and arm effect draws with window metadata.

Return type:

xr.Dataset