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

doc: comments

Browse Source
This commit is contained in:
Skudalen 2021-06-22 21:00:51 +02:00
parent 15937bdd64
commit 0fd756be6f
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Handle_emg_data.py
View File

@ -8,18 +8,22 @@ class CSV_handler:
def __init__(self): def __init__(self):
self.working_dir = str(Path.cwd()) self.working_dir = str(Path.cwd())
# Makes dataframe from the csv files in the working directory
def make_df(self, filename) def make_df(self, filename)
filepath = self.working_dir + str(filename) filepath = self.working_dir + str(filename)
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
def get_time_emg_table(self, filename, emg_nr): def get_time_emg_table(self, filename, emg_nr):
tot_data_frame = make_df(self, filename) tot_data_frame = make_df(self, filename)
emg_str = "emg" + str(emg_nr) emg_str = "emg" + str(emg_nr)
filtered_df = tot_data_frame[emg_str] filtered_df = tot_data_frame[["timestamp", emg_str]]
return filtered_df return filtered_df