From c650eb928cf7788bea8d6788215cd13163a819d3 Mon Sep 17 00:00:00 2001 From: Skudalen Date: Fri, 25 Jun 2021 14:00:25 +0200 Subject: [PATCH] doc: describe the get get_df func --- Handle_emg_data.py | 1 + Signal_prep.py | 5 ++--- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Handle_emg_data.py b/Handle_emg_data.py index f194ee9..695b678 100644 --- a/Handle_emg_data.py +++ b/Handle_emg_data.py @@ -443,6 +443,7 @@ class CSV_handler: self.data_type = 'soft' return self.data_container_dict + # Retrieves df via the data_dict in the handler object def get_df_from_data_dict(self, subject_nr, which_arm, round, emg_nr): data_type = self.data_type container = self.data_container_dict.get(subject_nr) diff --git a/Signal_prep.py b/Signal_prep.py index 2c98681..fc43388 100644 --- a/Signal_prep.py +++ b/Signal_prep.py @@ -71,9 +71,8 @@ def inverse_wavelet(df, cA_filt, cD_filt): return y_new_values # Takes in handler and detailes to denoise. Returns arrays and df -def denoice_dataset(handler:Handler.CSV_handler, subject_nr, which_arm, emg_nr, round): - - print(df.head) +def denoice_dataset(handler:Handler.CSV_handler, subject_nr, which_arm, round, emg_nr): + df = handler.get_df_from_data_dict(subject_nr, which_arm, round, emg_nr) N = get_xory_from_df('x', df) N_trans, cA, cD = wavelet_db4_denoising(df)