diffindiff 2.5.0__tar.gz → 2.5.1__tar.gz

{diffindiff-2.5.0 → diffindiff-2.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffindiff
-Version: 2.5.0
+Version: 2.5.1
 Summary: diffindiff: Python library for convenient Difference-in-Differences analyses
 Author: Thomas Wieland
 Author-email: geowieland@googlemail.com
@@ -29,7 +29,7 @@ A research note featuring a case study that utilizes the diffindiff library is a
 
 If you use this software, please cite:
 
-Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.5.0) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
+Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.5.1) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -177,16 +177,9 @@ See the /tests directory for usage examples of most of the included functions.
 This software was developed without the use of AI-generated code. The Continue Agent in Microsoft Visual Studio Code using the GPT-5 mini model (by OpenAI) was used solely to assist in drafting and refining docstrings for documentation. The corresponding guidelines and constraints defined by the author are documented in `AGENTS-docstrings.md` in the [public GitHub repository](https://github.com/geowieland/diffindiff_official).
 
 
-## What's new (v2.5.0)
+## What's new (v2.5.1)
 
-- General
-  - Update of dependencies to be compatible with Python 3.13 (still works with Python 3.11) while avoiding incompatibility conflict of statsmodels and scipy
-  - Internal changes in some functions to be compatible with Python 3.13 (still works with Python 3.11)
-- Extensions
-  - Creating synthetic control units with DiffData.add_synthetic() and conducting a Synthetic DiD analysis
-  - didanalysis_helper.treatment_diagnostics() now additionally checks unique number of analysis units and time points
-  - didanalysis_helper.data_diagnostics() now additionally checks whether covariates are constant; didanalysis.did_analysis() and .ddd_analysis() automatically skip such variables from the model analysis
 - Bugfixes
-  - DiffData.add_own_counterfactual() now works correctly in any case (also when no. of treatments > 1)
-  - Correct internal processing of treatment data in DiffTreatment and DiffData objects when treatments were added
-  - Fixed missing treatment group definition in DiffData.add_treatment() 
+  - In DiffData.create_treatment() and DiffData.create_data(), pre-post designs with after-treatment period (follow-up) now work correctly
+  - DiffModel.plot() nows supports pre-post designs with after-treatment period
+  - Some small stabilizations

{diffindiff-2.5.0 → diffindiff-2.5.1}/README.md

@@ -21,7 +21,7 @@ A research note featuring a case study that utilizes the diffindiff library is a
 
 If you use this software, please cite:
 
-Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.5.0) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
+Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.5.1) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -169,16 +169,9 @@ See the /tests directory for usage examples of most of the included functions.
 This software was developed without the use of AI-generated code. The Continue Agent in Microsoft Visual Studio Code using the GPT-5 mini model (by OpenAI) was used solely to assist in drafting and refining docstrings for documentation. The corresponding guidelines and constraints defined by the author are documented in `AGENTS-docstrings.md` in the [public GitHub repository](https://github.com/geowieland/diffindiff_official).
 
 
-## What's new (v2.5.0)
+## What's new (v2.5.1)
 
-- General
-  - Update of dependencies to be compatible with Python 3.13 (still works with Python 3.11) while avoiding incompatibility conflict of statsmodels and scipy
-  - Internal changes in some functions to be compatible with Python 3.13 (still works with Python 3.11)
-- Extensions
-  - Creating synthetic control units with DiffData.add_synthetic() and conducting a Synthetic DiD analysis
-  - didanalysis_helper.treatment_diagnostics() now additionally checks unique number of analysis units and time points
-  - didanalysis_helper.data_diagnostics() now additionally checks whether covariates are constant; didanalysis.did_analysis() and .ddd_analysis() automatically skip such variables from the model analysis
 - Bugfixes
-  - DiffData.add_own_counterfactual() now works correctly in any case (also when no. of treatments > 1)
-  - Correct internal processing of treatment data in DiffTreatment and DiffData objects when treatments were added
-  - Fixed missing treatment group definition in DiffData.add_treatment() 
+  - In DiffData.create_treatment() and DiffData.create_data(), pre-post designs with after-treatment period (follow-up) now work correctly
+  - DiffModel.plot() nows supports pre-post designs with after-treatment period
+  - Some small stabilizations

{diffindiff-2.5.0 → diffindiff-2.5.1}/diffindiff/config.py

@@ -4,15 +4,15 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     1.0.21
-# Last update: 2026-07-03 17:50
+# Version:     1.0.22
+# Last update: 2026-07-10 11:36
 # Copyright (c) 2025-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
 # Basic config:
 
 PACKAGE_NAME = "diffindiff"
-PACKAGE_VERSION = "2.5.0"
+PACKAGE_VERSION = "2.5.1"
 
 VERBOSE = False
 
@@ -52,6 +52,7 @@ TREATMENT_PERIOD_DESCRIPTION = f"{TREATMENT_DESCRIPTION} period"
 STUDY_PERIOD_DESCRIPTION = "Study period"
 PREPOST_DESCRIPTION = "Pre-post"
 AFTER_TREATMENT_PERIOD_DESCRIPTION = "After-treatment period"
+FOLLOW_UP_DESCRIPTION = "Follow-up"
 
 UNITS_DESCRIPTION = "Units"
 

{diffindiff-2.5.0 → diffindiff-2.5.1}/diffindiff/didanalysis.py

@@ -4,8 +4,8 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     2.4.1
-# Last update: 2026-07-02 20:17
+# Version:     2.4.2
+# Last update: 2026-07-10 11:30
 # Copyright (c) 2024-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
@@ -898,6 +898,49 @@ class DiffModel:
         
         treatment_effects = self.treatment_effects()
 
+        try:
+            if model_config.get("pre_post", False) and len(model_config.get("after_treatment_col", [])) > 0:
+
+                labels = treatment_effects.iloc[:, 0].astype(str).values
+                has_after = any((config.AFTER_TREATMENT_PERIOD_DESCRIPTION in lab) or (config.ATT_COL in lab) for lab in labels)
+                if not has_after:
+                    model_results = self.data[0]
+
+                    aate_key = config.EFFECTS_TYPES["AATE"]["model_results_key"] if "AATE" in config.EFFECTS_TYPES else "average_after_treatment_effects"
+                    att_key = config.EFFECTS_TYPES["ATT"]["model_results_key"] if "ATT" in config.EFFECTS_TYPES else "after_treatment_time_effects"
+                    appended = False
+
+                    for key_try in [aate_key, att_key]:
+                        if key_try in model_results and isinstance(model_results[key_try], dict):
+                            for subkey, val in model_results[key_try].items():
+
+                                if isinstance(val, dict):
+
+                                    coef_k = config.OLS_MODEL_RESULTS["coef"]["model_results_key"]
+                                    se_k = config.OLS_MODEL_RESULTS["coef_standard_errors"]["model_results_key"]
+                                    t_k = config.OLS_MODEL_RESULTS["coef_teststatistic"]["model_results_key"]
+                                    p_k = config.OLS_MODEL_RESULTS["coef_p"]["model_results_key"]
+                                    cil_k = config.OLS_MODEL_RESULTS["coef_confint_lower"]["model_results_key"]
+                                    ciu_k = config.OLS_MODEL_RESULTS["coef_confint_upper"]["model_results_key"]
+
+                                    new_row = {
+                                        "": f"{config.EFFECTS_TYPES.get('AATE', {'description': 'After-treatment'})['description']} {subkey}",
+                                        "Estimate": val.get(coef_k, np.nan),
+                                        "SE": val.get(se_k, np.nan),
+                                        "t": val.get(t_k, np.nan),
+                                        "p": val.get(p_k, np.nan),
+                                        "CI lower": val.get(cil_k, np.nan),
+                                        "CI upper": val.get(ciu_k, np.nan),
+                                    }
+                                    treatment_effects = pd.concat([treatment_effects, pd.DataFrame([new_row])], ignore_index=True)
+                                    appended = True
+                                    break
+                        if appended:
+                            break
+        except Exception:
+
+            pass
+
         if sort_by_coef:
             treatment_effects = treatment_effects.sort_values(
                 by = treatment_effects.columns[1],
@@ -1807,14 +1850,14 @@ class DiffModel:
         model_data_TG_CG["t"] = pd.to_datetime(model_data_TG_CG["t"])
 
         if not model_config["pre_post"]:
-            pre_post_barplot = False        
+            pre_post_barplot = False
 
         if pre_post_barplot:
 
             x_pos_t1_TG = 0
             x_pos_t1_CG = x_pos_t1_TG + pre_post_bar_width  
             x_pos_t2_TG = 1.5  
-            x_pos_t2_CG = x_pos_t2_TG + pre_post_bar_width  
+            x_pos_t2_CG = x_pos_t2_TG + pre_post_bar_width
 
             plt.bar(
                 x = x_pos_t1_TG, 
@@ -1842,6 +1885,24 @@ class DiffModel:
                 color=lines_col[3], 
                 width = pre_post_bar_width
                 )
+            
+            if len(model_data_TG_CG) == 3:
+
+                x_pos_t3_TG = 3  
+                x_pos_t3_CG = x_pos_t3_TG + pre_post_bar_width
+
+                plt.bar(
+                    x = x_pos_t3_TG, 
+                    height = model_data_TG_CG[outcome_col_expected_TG][2],                 
+                    color = lines_col[2], 
+                    width = pre_post_bar_width
+                    )            
+                plt.bar(
+                    x = x_pos_t3_CG, 
+                    height = model_data_TG_CG[outcome_col_expected_CG][2],                 
+                    color=lines_col[3], 
+                    width = pre_post_bar_width
+                    )
 
             plt.xlabel(x_label)
             plt.ylabel(y_label)
@@ -1935,16 +1996,29 @@ class DiffModel:
             ax.xaxis.set_major_formatter(DateFormatter(model_config["date_format"]))
 
         if model_config["pre_post"]:
+
+            ticks_line = model_data_TG_CG["t"].unique()
+            ticks_bars = [0.25, 1.75]
+
+            if len(model_data_TG_CG) == 3:
+                
+                if len(pre_post_ticks) < 3:
+                    pre_post_ticks.append("Follow-up")
+
+                ticks_bars.append(3.25)
+
             if not pre_post_barplot:
+                
                 plt.xticks(
-                    model_data_TG_CG["t"].unique(), 
-                    labels = [pre_post_ticks[0], pre_post_ticks[1]]
+                    ticks_line, 
+                    labels = pre_post_ticks
                     )  
             else:
                 plt.xticks(
-                    [0.25, 1.75], 
-                    labels = [pre_post_ticks[0], pre_post_ticks[1]]
-                    )  
+                    ticks_bars, 
+                    labels = pre_post_ticks
+                    )
+                
         else:
             plt.xticks(rotation=90)
         
@@ -2644,6 +2718,8 @@ def did_analysis(
     FE_time_vars = []
     dummy_time_original = []
     
+    drop_first = intercept
+
     if not demean:
 
         if FE_unit:
@@ -2652,7 +2728,7 @@ def did_analysis(
                 data = data,
                 col = unit_col,
                 type = "unit",
-                drop_first = intercept,
+                drop_first = drop_first,
                 verbose = verbose
             )
             
@@ -2667,7 +2743,7 @@ def did_analysis(
                 data = data,
                 col = time_col,
                 type = "time",
-                drop_first = intercept,
+                drop_first = drop_first,
                 verbose = verbose
             )
             
@@ -2691,7 +2767,7 @@ def did_analysis(
                 data = data,
                 col = group_by,
                 type = "group",
-                drop_first = intercept,
+                drop_first = drop_first,
                 verbose = verbose
                 )
         

{diffindiff-2.5.0 → diffindiff-2.5.1}/diffindiff/diddata.py

@@ -4,8 +4,8 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     2.3.0
-# Last update: 2026-07-03 18:00
+# Version:     2.3.1
+# Last update: 2026-07-10 11:37
 # Copyright (c) 2024-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
@@ -622,15 +622,36 @@ def create_treatment(
 
     if pre_post:
 
-        after_treatment_period = False
+        if after_treatment_period:
+
+            if study_period[1] <= treatment_period[1]:
+                
+                print(f"WARNING: after_treatment_period is set True, but end of study period ({study_period[1]}) is <= end of treatment period ({treatment_period[1]}). After treatment period is ignored.")
+                
+                after_treatment_period = False
+
+            else:
+
+                study_period_range = [study_period[0], study_period[1]]
+                study_period_N = 3
+                study_period_counter = [1,2,3]
 
-        study_period_range = [study_period[0], study_period[1]]
-        study_period_N = 2
-        study_period_counter = [1,2]
+                treatment_period_N = 1
+                TT_dummies = [0,1,0]
+                ATT_dummies = [0,0,1]
 
-        treatment_period_range = [treatment_period[0], treatment_period[1]]
-        treatment_period_N = 1
-        TT_dummies = [0,1]
+                treatment_period_range = [study_period[0], treatment_period[1], study_period[1]]
+                after_treatment_period_range = treatment_period_range
+        
+        else:
+                
+            study_period_range = [study_period[0], study_period[1]]
+            study_period_N = 2
+            study_period_counter = [1,2]
+
+            treatment_period_range = [treatment_period[0], treatment_period[1]]
+            treatment_period_N = 1
+            TT_dummies = [0,1]
 
         study_period_range = pd.DataFrame (treatment_period_range, columns=[config.TIME_COL])
         study_period_range[config.TIME_COUNTER_COL] = pd.DataFrame(study_period_counter)
@@ -640,7 +661,7 @@ def create_treatment(
             TT_col: TT_dummies
             }
         
-        TT_data = pd.DataFrame(TT_data)
+        TT_data = pd.DataFrame(TT_data)    
         
         treatment_period_range = pd.DataFrame(
             study_period_range
@@ -650,6 +671,24 @@ def create_treatment(
             TT_data, 
             how = "left"
             )
+        
+        if after_treatment_period:
+
+            ATT_data = {
+                config.TIME_COL: after_treatment_period_range, 
+                ATT_col: ATT_dummies
+                }
+
+            ATT_data = pd.DataFrame(ATT_data)
+
+            after_treatment_period_range = pd.DataFrame (after_treatment_period_range, columns=[config.TIME_COL])
+            after_treatment_period_range[config.TIME_COUNTER_COL] = pd.DataFrame(study_period_counter)
+
+            after_treatment_data_df = after_treatment_period_range.merge(ATT_data, how = "left")
+            after_treatment_data_df[ATT_col] = after_treatment_data_df[ATT_col].fillna(0)
+            after_treatment_data_df = after_treatment_data_df.drop(columns=[config.TIME_COL, config.TIME_COUNTER_COL])
+
+            treatment_data_df = pd.concat([treatment_data_df, after_treatment_data_df], axis=1)
 
     else:
 
@@ -689,7 +728,7 @@ def create_treatment(
 
     treatment_data_df[TT_col] = treatment_data_df[TT_col].fillna(0)
 
-    if after_treatment_period:
+    if after_treatment_period and not pre_post:
         
         treatment_period_last = datetime.strptime(
             treatment_period[1], 
@@ -720,6 +759,9 @@ def create_treatment(
 
     else:
         after_treatment_period_N = 0
+
+    if after_treatment_period and pre_post:
+        after_treatment_period_N = 1
     
     no_treatments = 1
 
@@ -735,7 +777,7 @@ def create_treatment(
         "date_format": date_format,
         "pre_post": pre_post
         }
-
+    
     treatment_config = {
         0: {
             "treatment_name": treatment_name,
@@ -2050,7 +2092,10 @@ class DiffData:
             else:
                 print(f"  {config.STUDY_PERIOD_DESCRIPTION:<{config.DIDDATA_SUMMARY_MAX_WIDTH}} {treatment_meta['study_period_start']} - {treatment_meta['study_period_end']} ({treatment_meta['study_period']} {treatment_meta['frequency']})")
                 print(f"  {config.TREATMENT_PERIOD_DESCRIPTION:<{config.DIDDATA_SUMMARY_MAX_WIDTH}} {treatment_config[key]['treatment_period_start']} - {treatment_config[key]['treatment_period_end']} ({treatment_config[key]['treatment_period']} {treatment_meta['frequency']})")
-                
+
+            if value["after_treatment_period"]:
+                print(f"  {config.AFTER_TREATMENT_PERIOD_DESCRIPTION:<{config.DIDDATA_SUMMARY_MAX_WIDTH}} {treatment_config[key]['treatment_period_end']} vs. {treatment_meta['study_period_end']} ({config.FOLLOW_UP_DESCRIPTION})")
+
         print("-" * total_width)
 
         print(f"Outcome '{outcome_col_original}'")

{diffindiff-2.5.0 → diffindiff-2.5.1}/diffindiff/tests/tests_diffindiff.py

@@ -4,8 +4,8 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     2.0.15
-# Last update: 2026-05-01 09:54
+# Version:     2.1.0
+# Last update: 2026-07-10 11:35
 # Copyright (c) 2025-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
@@ -75,7 +75,7 @@ curfew_data_prepost=create_data(
     treatment_period=["2020-03-21", "2020-05-05"],
     freq="D",
     pre_post=True,
-    after_treatment_period = True
+    after_treatment_period = False
     )
 # Creating DiD treatement dataset by defining groups and
 # treatment time at once
@@ -98,6 +98,9 @@ print(curfew_model_prepost.covariates())
 curfew_model_prepost.summary()
 # Model summary
 
+print(curfew_model_prepost.didmodel().summary())
+# Access statsmodels OLS object
+
 print(curfew_model_prepost.fit_metrics())
 # Show model fit metrics
 
@@ -137,6 +140,59 @@ curfew_model_prepost.plot_treatment_effects(
     )
 # plot effects
 
+
+# Same analysis with after-treatment period (follow-up):
+
+curfew_data_prepost_AT=create_data(
+    outcome_data=curfew_DE,
+    unit_id_col="county",
+    time_col="infection_date",
+    outcome_col="infections_cum_per100000",
+    treatment_group= 
+        curfew_DE.loc[curfew_DE["Bundesland"].isin([9,10,14])]["county"],
+    control_group= 
+        curfew_DE.loc[~curfew_DE["Bundesland"].isin([9,10,14])]["county"],
+    study_period=["2020-03-01", "2020-05-15"],
+    treatment_period=["2020-03-21", "2020-05-05"],
+    freq="D",
+    pre_post=True,
+    after_treatment_period = True
+    )
+# Creating DiD treatement dataset by defining groups and
+# treatment time at once
+
+curfew_data_prepost_AT.summary()
+# Summary of created data
+
+curfew_model_prepost_AT=curfew_data_prepost_AT.analysis(verbose=True)
+# Model analysis of created data
+
+curfew_model_prepost_AT.summary()
+# Model summary
+
+curfew_model_prepost_AT.plot(
+    x_label="Timepoint",
+    y_label="Cumulative infections per 100,000",
+    plot_title="Curfew effectiveness pre-post - Groups over time",
+    plot_observed=False,
+    lines_col=[None,None,"blue","orange"],
+    lines_labels=[None,None,"Treatment group","Control group","Treatment group CI","Control group CI"],
+    lines_style=[None,None,"solid","solid"]
+    )
+# Plot DiD pre vs. post results
+# with user-determined style
+
+curfew_model_prepost_AT.plot(
+    x_label="Timepoint",
+    y_label="Cumulative infections per 100,000",
+    plot_title="Curfew effectiveness pre-post - Groups over time",
+    lines_col=[None,None,"blue","orange"],
+    lines_labels=[None,None,"Treatment group","Control group","Treatment group CI", "Control group CI"],
+    pre_post_barplot=True
+    )
+# Plot DiD pre vs. post results
+# with user-determined style
+
 counties_DE=pd.read_csv("data/counties_DE.csv", sep=";", decimal=",", encoding='latin1')
 # Dataset with German county data
 

{diffindiff-2.5.0 → diffindiff-2.5.1}/diffindiff.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffindiff
-Version: 2.5.0
+Version: 2.5.1
 Summary: diffindiff: Python library for convenient Difference-in-Differences analyses
 Author: Thomas Wieland
 Author-email: geowieland@googlemail.com
@@ -29,7 +29,7 @@ A research note featuring a case study that utilizes the diffindiff library is a
 
 If you use this software, please cite:
 
-Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.5.0) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
+Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.5.1) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -177,16 +177,9 @@ See the /tests directory for usage examples of most of the included functions.
 This software was developed without the use of AI-generated code. The Continue Agent in Microsoft Visual Studio Code using the GPT-5 mini model (by OpenAI) was used solely to assist in drafting and refining docstrings for documentation. The corresponding guidelines and constraints defined by the author are documented in `AGENTS-docstrings.md` in the [public GitHub repository](https://github.com/geowieland/diffindiff_official).
 
 
-## What's new (v2.5.0)
+## What's new (v2.5.1)
 
-- General
-  - Update of dependencies to be compatible with Python 3.13 (still works with Python 3.11) while avoiding incompatibility conflict of statsmodels and scipy
-  - Internal changes in some functions to be compatible with Python 3.13 (still works with Python 3.11)
-- Extensions
-  - Creating synthetic control units with DiffData.add_synthetic() and conducting a Synthetic DiD analysis
-  - didanalysis_helper.treatment_diagnostics() now additionally checks unique number of analysis units and time points
-  - didanalysis_helper.data_diagnostics() now additionally checks whether covariates are constant; didanalysis.did_analysis() and .ddd_analysis() automatically skip such variables from the model analysis
 - Bugfixes
-  - DiffData.add_own_counterfactual() now works correctly in any case (also when no. of treatments > 1)
-  - Correct internal processing of treatment data in DiffTreatment and DiffData objects when treatments were added
-  - Fixed missing treatment group definition in DiffData.add_treatment() 
+  - In DiffData.create_treatment() and DiffData.create_data(), pre-post designs with after-treatment period (follow-up) now work correctly
+  - DiffModel.plot() nows supports pre-post designs with after-treatment period
+  - Some small stabilizations

{diffindiff-2.5.0 → diffindiff-2.5.1}/setup.py

@@ -7,7 +7,7 @@ def read_README():
     
 setup(
     name='diffindiff',
-    version='2.5.0',
+    version='2.5.1',
     description='diffindiff: Python library for convenient Difference-in-Differences analyses',
     packages=find_packages(include=["diffindiff", "diffindiff.tests"]),
     include_package_data=True,