ms762wu/EMG_Biometrics_2021
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

feat: get_time_emg_table puts the df into the subjects data_container

Browse Source
This commit is contained in:
Skudalen 2021-06-23 11:41:06 +02:00
parent bca3dbc9b4
commit fee616d7d6
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Handle_emg_data.py
View File

@ -8,7 +8,7 @@ class Data_container:
def __init__(self, subject_nr:int, subject_name:str): def __init__(self, subject_nr:int, subject_name:str):
self.subject_nr = subject_nr self.subject_nr = subject_nr
self.subject_name = subject_name self.subject_name = subject_name
self.data_dict = {'left': [], 'right': []} self.data_dict = {'left': [None]*8, 'right': [None]*8}
class CSV_handler: class CSV_handler:
@ -28,8 +28,12 @@ class CSV_handler:
emg_str = 'emg' + str(emg_nr) emg_str = 'emg' + str(emg_nr)
filtered_df = tot_data_frame[["timestamp", emg_str]] filtered_df = tot_data_frame[["timestamp", emg_str]]
# Links the retrieved data with the subjects data_container
subject_nr = data_container.subject_nr subject_nr = data_container.subject_nr
self.data_container_dict[subject_nr] = data_container self.data_container_dict[subject_nr] = data_container
# Places the data correctly:
if which_arm is 'left':
data_container.data_dict['left'][emg_nr+1] = filtered_df
return filtered_df return filtered_df