Merge pull request #8 from UDST/agg_max_min_

added min and max to the agg function
UDST · Aug 4, 2020 · eccd7a1 · eccd7a1
2 parents 4d2a57e + 28bb39b
commit eccd7a1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/variable_generators/generators.py b/variable_generators/generators.py
@@ -35,6 +35,10 @@ def func():
             values = groupby.std().fillna(0)
         if agg_function == 'sum':
             values = groupby.sum().fillna(0)
+        if agg_function == 'max':
+            values = groupby.max().fillna(0)
+        if agg_function == 'min':
+            values = groupby.min().fillna(0)
 
         locations_index = orca.get_table(geog).index
         series = pd.Series(data=values, index=locations_index)