oups 2025.9.5__py3-none-any.whl

oups/stateful_ops/aggstream/utils.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+"""
+Created on Wed Jan 24 21:30:00 2024.
+
+@author: pierrot
+
+"""
+import operator
+
+from numpy import ones
+from numpy import zeros
+from numpy.typing import NDArray
+from pandas import DataFrame
+
+
+ops = {
+    "==": operator.eq,
+    "=": operator.eq,
+    "!=": operator.ne,
+    ">": operator.gt,
+    ">=": operator.ge,
+    "<": operator.lt,
+    "<=": operator.le,
+}
+
+
+def dataframe_filter(df: DataFrame, filters) -> NDArray:
+    """
+    Produce a column filter of the input dataframe.
+
+    Parameters
+    ----------
+    df : DataFrame
+        DataFrame to filter.
+    filters : list[list[tuple]]
+       To filter out data from seed.
+       Filter syntax: [[(column, op, val), ...],...]
+       where op is [==, =, >, >=, <, <=, !=, in, not in, ~]
+       The innermost tuples are transposed into a set of filters applied
+       through an `AND` operation.
+       The outer list combines these sets of filters through an `OR` operation.
+       A single list of tuples can also be used, meaning that no `OR` operation
+       between set of filters is to be conducted.
+
+    Returns
+    -------
+    numpy 1D-array
+       Array of rows to keep (boolean mask).
+
+    """
+    if isinstance(filters[0], tuple):
+        raise ValueError(
+            "not possible to have a 'filters' parameter without at least an inner list.",
+        )
+    out = zeros(len(df), dtype=bool)
+    for or_part in filters:
+        and_part = ones(len(df), dtype=bool)
+        for name, op, val in or_part:
+            if op == "in":
+                and_part &= df[name].isin(val).values
+            elif op == "not in":
+                and_part &= ~df[name].isin(val).values
+            elif op in ops:
+                and_part &= ops[op](df[name].values, val)
+            elif op == "~":
+                and_part &= ~df[name].values
+            else:
+                # Unknown operator.
+                raise ValueError(f"operator '{op}' is not supported.")
+        out |= and_part
+    return out

oups/stateful_ops/asof_merger/__init__.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python3
+"""
+Created on Sat Jun 28 18:35:00 2025.
+
+@author: pierrot
+
+"""
+from .asof_merger import AsofMerger
+
+
+__all__ = ["AsofMerger"]