From 1bdb334c42de32f9d5bcc0806e1be7488c9b03e4 Mon Sep 17 00:00:00 2001 From: junos Date: Tue, 27 Jul 2021 20:57:21 +0200 Subject: [PATCH] Fix formatting. --- features/esm_SAM.py | 23 +++++++++++++++++------ 1 file changed, 17 insertions(+), 6 deletions(-) diff --git a/features/esm_SAM.py b/features/esm_SAM.py index 0b0ec42..dd81404 100644 --- a/features/esm_SAM.py +++ b/features/esm_SAM.py @@ -104,14 +104,25 @@ def calculate_threat_challenge_means(df_esm_sam_clean: pd.DataFrame) -> pd.DataF ) ] # Calculate mean of threat and challenge subscales for each ESM session. - df_esm_event_threat_challenge_mean_wide = pd.pivot_table(df_esm_event_threat_challenge, index=["participant_id","device_id", "esm_session"], columns=["questionnaire_id"], values=["esm_user_answer_numeric"], aggfunc="mean") + df_esm_event_threat_challenge_mean_wide = pd.pivot_table( + df_esm_event_threat_challenge, + index=["participant_id", "device_id", "esm_session"], + columns=["questionnaire_id"], + values=["esm_user_answer_numeric"], + aggfunc="mean", + ) # Drop unnecessary column values. - df_esm_event_threat_challenge_mean_wide.columns = df_esm_event_threat_challenge_mean_wide.columns.get_level_values(1) + df_esm_event_threat_challenge_mean_wide.columns = df_esm_event_threat_challenge_mean_wide.columns.get_level_values( + 1 + ) df_esm_event_threat_challenge_mean_wide.columns.name = None - df_esm_event_threat_challenge_mean_wide.rename(columns={ - QUESTIONNAIRE_ID_SAM.get("event_threat"): "threat_mean", - QUESTIONNAIRE_ID_SAM.get("event_challenge"): "challenge_mean" - }, inplace=True) + df_esm_event_threat_challenge_mean_wide.rename( + columns={ + QUESTIONNAIRE_ID_SAM.get("event_threat"): "threat_mean", + QUESTIONNAIRE_ID_SAM.get("event_challenge"): "challenge_mean", + }, + inplace=True, + ) return df_esm_event_threat_challenge_mean_wide