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)
 #########################