Mercurial Hosting > traffic-intelligence
diff trafficintelligence/utils.py @ 1058:16575ca4537d
work in progress
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Tue, 10 Jul 2018 17:16:38 -0400 |
parents | 4069d8545922 |
children | a87b3072bd26 |
line wrap: on
line diff
--- a/trafficintelligence/utils.py Mon Jul 09 16:21:03 2018 -0400 +++ b/trafficintelligence/utils.py Tue Jul 10 17:16:38 2018 -0400 @@ -591,6 +591,19 @@ print('Unknown aggregation method: {}'.format(funcStr)) return None +def aggregationMethods(methods, centiles = None): + aggFunctions = {} + headers = [] + for method in methods: + if method == 'centile': + aggFunctions[method] = aggregationFunction(method, centiles) + for c in centiles: + headers.append('{}{}'.format(method,c)) + else: + aggFunctions[method] = utils.aggregationFunction(method) + headers.append(method) + return aggFunctions, headers + ######################### # regression analysis using statsmodels (and pandas) #########################