sander/rowsandall
Private
Public Access
1
0
Fork 0
Code Activity

bug fix dataprepnodjango

Browse Source
This commit is contained in:
Sander Roosendaal
2019-11-06 16:07:38 +01:00
parent 89f607ec8b
commit 50ac712bc2
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rowers/dataprep.py
View File

@@ -2534,9 +2534,11 @@ def dataprep(rowdatadf, id=0, bands=True, barchart=True, otwpower=True,
if id != 0: if id != 0:
data['workoutid'] = id data['workoutid'] = id
data.fillna(0,inplace=True) data.fillna(0,inplace=True)
data = data.astype( for k, v in dtypes.items():
dtype=dtypes, try:
) data[k] = data[k].astype(v)
except KeyError:
pass
filename = 'media/strokedata_{id}.parquet.gz'.format(id=id) filename = 'media/strokedata_{id}.parquet.gz'.format(id=id)

8
rowers/dataprepnodjango.py
View File

@@ -1254,7 +1254,13 @@ def dataprep(rowdatadf,id=0,bands=True,barchart=True,otwpower=True,
# write data if id given # write data if id given
if id != 0: if id != 0:
data['workoutid'] = id data['workoutid'] = id
data = data.astype(dtype=dtypes) data.fillna(0,inplace=True)
for k, v in dtypes.items():
try:
data[k] = data[k].astype(v)
except KeyError:
pass
filename = 'media/strokedata_{id}.parquet.gz'.format(id=id) filename = 'media/strokedata_{id}.parquet.gz'.format(id=id)
df = dd.from_pandas(data,npartitions=1) df = dd.from_pandas(data,npartitions=1)
df.to_parquet(filename,engine='fastparquet',compression='GZIP') df.to_parquet(filename,engine='fastparquet',compression='GZIP')