PreProcessPy/preSDK.py at master · Neuro-UOM/PreProcessPy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# -*- coding: utf-8 -*-

from matplotlib import pyplot as plt
import seaborn as sns
import os
import glob
import numpy as np
import pandas as pd

def createCSV(path1,name):
    path = path1

    allFiles = glob.glob(path + "/*.csv")
    frame = pd.DataFrame()
    list_ = []
    for file_ in allFiles:
        df = pd.read_csv(file_,index_col=None, header=0)
        list_.append(df)
    frame = pd.concat(list_)


    features = []
    waves = ["Low_beta","High_beta","Alpha","Theta", "Gamma"]
    for i in range(7,13):
        for j in waves:
            features.append(str(i)+ " "+ j)

    features.append("Label")
    print features

    frame = frame[features]
    frame['Label'] = frame['Label'].map({'null': 2, 'green': 1, 'red': 0})

    frame.to_csv(name)

createCSV('./Data/All','All.csv')
createCSV('./Data/Sala','Sala.csv')