From a2180aee544bdc2c4509f7b1c4d4e9c19719a835 Mon Sep 17 00:00:00 2001 From: junos Date: Wed, 11 Aug 2021 14:53:59 +0200 Subject: [PATCH] Fix assignment to use loc. For assigning a value to selected rows (a subset), regular slicing using [] produces a KeyError. --- features/esm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/features/esm.py b/features/esm.py index 88318fa..67635b8 100644 --- a/features/esm.py +++ b/features/esm.py @@ -256,9 +256,9 @@ def clean_up_esm(df_esm_preprocessed: pd.DataFrame) -> pd.DataFrame: ESM.ESM_TYPE.get("scale"), ESM.ESM_TYPE.get("number"), ] - df_esm_clean[df_esm_clean["esm_type"].isin(esm_type_numeric)] = df_esm_clean[ + df_esm_clean.loc[ df_esm_clean["esm_type"].isin(esm_type_numeric) - ].assign( + ] = df_esm_clean.loc[df_esm_clean["esm_type"].isin(esm_type_numeric)].assign( esm_user_answer_numeric=lambda x: x.esm_user_answer.str.slice(stop=1).astype( int )