Fixing pandas copy on slice warning

ONSdigital · May 17, 2024 · f1abca8 · f1abca8
1 parent f1372f0
commit f1abca8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/flag_and_count_matched_pairs.py b/src/flag_and_count_matched_pairs.py
@@ -39,7 +39,7 @@ def flag_matched_pair_merge(
         time_difference = -time_difference
 
     # Creating new DF, shifting period for forward or backward
-    df_with_predictive_column = df[[reference, strata, target]]
+    df_with_predictive_column = df.copy()[[reference, strata, target]]
     df_with_predictive_column["predictive_period"] = df[period] + pd.DateOffset(
         months=time_difference
     )