to_screen

Author: aidanacquah

src/actinet/accPlot.py

@@ -80,7 +80,7 @@ def main():  # noqa: C901
     if len(sys.argv) < 2:
         msg = "\nInvalid input, please enter at least 1 parameter, e.g."
         msg += "\npython accPlot.py timeSeries.csv.gz \n"
-        utils.toScreen(msg)
+        utils.to_screen(msg)
         parser.print_help()
         sys.exit(-1)
     args = parser.parse_args()

src/actinet/utils/utils.py

@@ -7,14 +7,17 @@
 from typing import Union
 
 
-def infer_freq(x):
+def infer_freq(t):
     """Like pd.infer_freq but more forgiving"""
-    freq, _ = stats.mode(np.diff(x), keepdims=False)
+    tdiff = t.to_series().diff()
+    q1, q3 = tdiff.quantile([0.25, 0.75])
+    tdiff = tdiff[(q1 <= tdiff) & (tdiff <= q3)]
+    freq = tdiff.mean()
     freq = pd.Timedelta(freq)
     return freq
 
 
-def toScreen(msg, verbose=True):
+def to_screen(msg, verbose=True):
     """
     Print msg str prepended with current time