Dendro

@schaefed

level1/do_dendro_calculation.py

@@ -25,7 +25,7 @@ def filter(data, flags):
     return data[data.columns[data.columns.str.match(".*dendro(?!.*r)")]]
 
 
-def grouper(df, n, fillvalue=None):
+def grouper(df, n=3, fillvalue=None):
     """
     This grouper function splits a pd.DataFrame into pd.DataFrames with `n` columns.
 
@@ -39,7 +39,7 @@ def grouper(df, n, fillvalue=None):
 
 
 def fillNa(df):
-    for bhd_value, bhd_manu, flag in grouper(df, n=3):
+    for bhd_value, bhd_manu, flag in grouper(df):
         # setting 91 flag on all values, where calculated value is nan
         df.loc[df[bhd_value].isna(), flag] = 91
         df[bhd_value] = df[bhd_value].fillna(df[bhd_manu])
@@ -139,7+139,7 @@
 def procDevice(logger, device, **kwargs):
     logger.info(f"processing: {device}")
 
     logger.debug("reading data")
     df = device.getL2Data(
         start_date=device.start_date,
         end_date=device.end_date,
@@ -182,7+182,7 @@
 
         # corrections in index Order and not setted flags
         final = final.sort_index()
-        for a, b, c in grouper(final, n=3):
+        for a, b, c in grouper(final):
             final = final.fillna({a: NODATA, b: NODATA, c: 92})
 
         writeData(final, device)