diff --git a/src/features/phone_esm/straw/process_user_event_related_segments.py b/src/features/phone_esm/straw/process_user_event_related_segments.py index 788d99ec..eae6fab3 100644 --- a/src/features/phone_esm/straw/process_user_event_related_segments.py +++ b/src/features/phone_esm/straw/process_user_event_related_segments.py @@ -84,7 +84,7 @@ def extract_ers(esm_df, device_id): .join(se_intensity, on=['device_id', 'esm_session'], how='inner') # Filter sessions that are not useful - extracted_ers = extracted_ers[(extracted_ers.se_time != "0 - Ne spomnim se") & (extracted_ers.se_duration != "0 - Ne spomnim se")] + extracted_ers = extracted_ers[(~extracted_ers.se_time.str.startswith("0 - ")) & (~extracted_ers.se_duration.str.startswith("0 - "))] # Transform data into its final form, ready for the extraction extracted_ers.reset_index(inplace=True) @@ -95,7 +95,7 @@ def extract_ers(esm_df, device_id): extracted_ers['se_duration'] = \ np.where( - extracted_ers['se_duration'] == "1 - Še vedno traja", + extracted_ers['se_duration'].str.startswith("1 - "), extracted_ers['session_end_timestamp'] - extracted_ers['event_timestamp'], extracted_ers['se_duration'] )