-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrecordingCSV.py
24 lines (20 loc) · 1.34 KB
/
recordingCSV.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import pandas as pd
import numpy as np
def savetoCSV(attention,meditation,delta,theta,lowAlpha,highAlpha,lowBeta,highBeta,lowGamma,highGamma,classification,filename):
try:
if classification==0:
classification=np.zeros(len(attention), dtype=int)
else:
classification=np.ones(len(attention), dtype=int)
dataset = pd.read_csv(filename+'.csv')
df = pd.DataFrame({"attention" : attention, "meditation" : meditation,"delta" : delta, "theta" : theta
, "lowAlpha" : lowAlpha, "highAlpha" : highAlpha, "lowBeta" : lowBeta,
"highBeta" : highBeta, "lowGamma" : lowGamma, "highGamma" : highGamma,"classification" : classification})
dataset=pd.concat([dataset,df])
dataset.to_csv(filename+'.csv',index=False)
# df.to_csv("meawake.csv", index=False)
except FileNotFoundError:
df = pd.DataFrame({"attention" : attention, "meditation" : meditation,"delta" : delta, "theta" : theta
, "lowAlpha" : lowAlpha, "highAlpha" : highAlpha, "lowBeta" : lowBeta,
"highBeta" : highBeta, "lowGamma" : lowGamma, "highGamma" : highGamma, "classification" : classification})
df.to_csv(filename+'.csv', index=False)