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

better metrics selection

Browse Source
This commit is contained in:
Sander Roosendaal
2019-11-18 10:52:34 +01:00
parent 9650a7cb71
commit 01c02b2b02
5 changed files with 66 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rowers/dataprep.py
View File

@@ -123,9 +123,11 @@ from scipy.signal import savgol_filter
import datetime
def get_video_data(w,groups=['basic']):
def get_video_data(w,groups=['basic'],mode='water'):
modes = [mode,'both','basic']
columns = ['time','velo','spm']
columns += [name for name,d in rowingmetrics if d['group'] in groups]
columns += [name for name,d in rowingmetrics if d['group'] in groups and d['mode'] in modes]
columns = list(set(columns))
df = getsmallrowdata_db(columns,ids=[w.id],
workstrokesonly=False,doclean=False,compute=False)