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)