-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdifference.py
More file actions
33 lines (24 loc) · 1.06 KB
/
Copy pathdifference.py
File metadata and controls
33 lines (24 loc) · 1.06 KB
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
import pandas as pd
import numpy as np
from glob import glob
files = glob( 'LabeledData/*.csv' )
saveData = []
saveTarget = []
for idx, file in enumerate( files ):
df = pd.read_csv( file, dtype=np.float32, index_col=None )
data = df.to_dict( 'list' )
dataNow = data[ 'Voltage' ][ 1: ]
dataLag = data[ 'Voltage' ][ :-1 ]
prepData = {}
prepData[ 'Time' ] = [ x - data[ 'Time' ][ 0 ] for x in data[ 'Time' ][ 1: ] ]
prepData[ 'Voltage' ] = [ x1 - x2 for ( x1, x2 ) in zip( dataNow, dataLag) ]
prepData[ 'Readout' ] = data[ 'Readout' ]
# Save in npz file
saveData.append( prepData[ 'Voltage' ] )
saveTarget.append( [ prepData[ 'Readout' ][ 0 ] ] )
# Save in csv file
name = "PreparedData\data_" + str( idx ) + ".csv"
pd.DataFrame( zip( prepData[ 'Time' ], prepData[ 'Voltage' ], prepData[ 'Readout' ] ), columns=[ 'Time', 'Voltage', 'Readout' ] ).to_csv( name, index=False )
data = np.stack( saveData, axis=0 )
target = np.stack( saveTarget, axis=0 )
np.savez( "ProcessedDataset.npz", data=saveData, label=saveTarget )