Skip to content
Snippets Groups Projects

Sc209/dropdown menus

Merged Ghost User requested to merge SC209/dropdown-menus into review-window
1 file
+ 20
7
Compare changes
  • Side-by-side
  • Inline
+ 20
7
import csv
import numpy as np
import pandas as pd
from abc import ABC, abstractmethod
@@ -13,6 +12,7 @@ class FileReader (ABC):
pass
@staticmethod
@abstractmethod
def extract_task_data(data):
"""Take the relevant data out of the imported data.
@@ -22,12 +22,7 @@ class FileReader (ABC):
Returns:
extraction: Dataframe of only the relevant data
"""
extraction = data.iloc[:, 5:8]
extraction = extraction.copy()
extraction.columns = ['Onset', 'Offset', 'Word']
extraction[['Onset', 'Offset']] = (extraction[['Onset', 'Offset']]
.apply(pd.to_numeric))
return extraction
pass
@staticmethod
def df_to_words(df, type_code='source'):
@@ -81,6 +76,24 @@ class CSVReader(FileReader):
words = self.df_to_words(data, type_code)
return words
@staticmethod
def extract_task_data(data):
"""Take the relevant data out of the already imported .csv
data.
Args:
data: Dataframe of imported .csv data
Returns:
extraction: Dataframe of only the relevant data
"""
extraction = data.iloc[:, 5:8]
extraction = extraction.copy()
extraction.columns = ['Onset', 'Offset', 'Word']
extraction[['Onset', 'Offset']] = (extraction[['Onset', 'Offset']]
.apply(pd.to_numeric))
return extraction
class FileWriter(ABC):
Loading