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

feat: connect data_container object to the retrieval of df's

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

21
Handle_emg_data.py
View File

@ -1,20 +1,20 @@
import pandas as pd import pandas as pd
from pathlib import Path from pathlib import Path
from pandas.core.frame import DataFrame
class Data_container: class Data_container:
def __init__(self): def __init__(self, subject_nr:int, subject_name:str):
self.subject_nr self.subject_nr = subject_nr
self.subject_name self.subject_name = subject_name
self.data_dict self.data_dict = {'left': [], 'right': []}
class CSV_handler: class CSV_handler:
def __init__(self): def __init__(self):
self.working_dir = str(Path.cwd()) self.working_dir = str(Path.cwd())
self.data_container_dict = {i: []} self.data_container_dict = {} # Dict with keys equal subject numbers and values equal the relvant datacontainer
# Makes dataframe from the csv files in the working directory # Makes dataframe from the csv files in the working directory
def make_df(self, filename): def make_df(self, filename):
@ -22,13 +22,14 @@ class CSV_handler:
df = pd.read_csv(filepath) df = pd.read_csv(filepath)
return df return df
# Extracts out the timestamp and the selected emg signal into a new dataframe # Extracts out the timestamp and the selected emg signal into a new dataframe and stores the data on the subject
def get_time_emg_table(self, filename: str, subject_nr: int, which_arm: str, emg_nr: int): def get_time_emg_table(self, filename:str, emg_nr:int, which_arm:str, data_container:Data_container):
tot_data_frame = self.make_df(filename) tot_data_frame = self.make_df(filename)
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]]
#self.data_dict[subject_nr] = [which_arm, emg1] subject_nr = data_container.subject_nr
self.data_container_dict[subject_nr] = data_container
return filtered_df return filtered_df