EMG_Biometrics_2021/Handle_emg_data.py

import pandas as pd
from pathlib import Path

class CSV_handler:

    def __init__(self):
        self.working_dir = str(Path.cwd())

    # Makes dataframe from the csv files in the working directory
    def make_df(self, filename):
        filepath = self.working_dir + str(filename)
        df = pd.read_csv(filepath)
        return df

    # Extracts out the timestamp and the selected emg signal into a new dataframe 
    def get_time_emg_table(self, filename, emg_nr):

        tot_data_frame = self.make_df(filename)
        emg_str = "emg" + str(emg_nr)
        filtered_df = tot_data_frame[["timestamp", emg_str]]
        return filtered_df

    
handler = CSV_handler()

file = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv"
subject1_left_emg1 = handler.get_time_emg_table(file, 1)
print(subject1_left_emg1.head)
#df = handler.make_df(file)
#print(df.head)