def findFeature(cat, feat): if cat == "time": return [f for f in tds.posFeatures().keys() if f in feat] if cat == "freq": return [f for f in fds.posFeatures().keys() if f in feat] if cat == "vel": return [f for f in vel.posFeatures().keys() if f in feat] if cat == "peak": return [f for f in ps.posFeatures().keys() if f in feat]
def getAllFeatures(): part = { 'time': { 'cols': tds.posCols(), 'features': tds.posFeatures().keys() }, 'freq': { 'cols': fds.posCols(), 'features': fds.posFeatures().keys() }, 'vel': { 'cols': vs.posCols(), 'features': vs.posFeatures().keys() }, 'peak': { 'cols': ps.posPeaks(), 'features': ps.posFeatures().keys() } } return { 'ankle': part, 'hip': part }