change ffill and bfill fro median

UDST · Nov 25, 2019 · d833cfb · d833cfb
1 parent 0ed6771
commit d833cfb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/variable_generators/generators.py b/variable_generators/generators.py
@@ -45,8 +45,7 @@ def func():
         if agg_function == 'sum':
             series = series.fillna(0)
         else:
-            series = series.fillna(method='ffill')
-            series = series.fillna(method='bfill')
+            series = series.fillna(series.median())
 
         return series