From 0e6a18a660507c7a50d0db1cd0b19164ab5f89d0 Mon Sep 17 00:00:00 2001 From: junos Date: Wed, 1 Dec 2021 16:48:17 +0100 Subject: [PATCH] Prepare data for export to CSV. --- participants/prepare_usernames_file.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/participants/prepare_usernames_file.py b/participants/prepare_usernames_file.py index 7614c46..16b03d9 100644 --- a/participants/prepare_usernames_file.py +++ b/participants/prepare_usernames_file.py @@ -1,5 +1,6 @@ import datetime import pandas as pd +from pyprojroot import here import participants.query_db @@ -25,3 +26,27 @@ participants_true_si = participants_overview_si[ participants_true_be = participants_overview_be[ participants_overview_be["SmartphoneBrand+Generation"].str.slice(0, 3) != "Not" ] + +# Concatenate participants from both countries. +participants_usernames_empatica = pd.concat( + [participants_true_be, participants_true_si] +) +# Filter only the participants from the main study (queried from the database). +participants_usernames_empatica = participants_usernames_empatica[ + participants_usernames_empatica["Username"].isin(participants_inactive_usernames) +] +# Rename and select columns. +participants_usernames_empatica = participants_usernames_empatica.rename( + columns={"Username": "label", "Wristband_SerialNo": "empatica_id"} +)[["label", "empatica_id"]] +# Adapt for csv export. +participants_usernames_empatica["empatica_id"] = participants_usernames_empatica[ + "empatica_id" +].str.replace(",", ";") + +participants_usernames_empatica.to_csv( + here("rapids/data/external/main_study_participants.csv"), + header=True, + index=False, + line_terminator="\n", +)