dycw-utilities 0.159.1__tar.gz → 0.159.2__tar.gz

{dycw_utilities-0.159.1 → dycw_utilities-0.159.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.159.1
+Version: 0.159.2
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.159.1 → dycw_utilities-0.159.2}/pyproject.toml

@@ -227,7 +227,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.159.1"
+version = "0.159.2"
 
 [project.entry-points.pytest11]
 pytest-randomly = "utilities.pytest_plugins.pytest_randomly"
@@ -259,7 +259,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.159.1"
+current_version = "0.159.2"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.159.1 → dycw_utilities-0.159.2}/src/tests/test_polars.py

@@ -164,6 +164,7 @@ from utilities.polars import (
     ensure_expr_or_series_many,
     expr_to_series,
     finite_ewm_mean,
+    first_true_horizontal,
     get_data_type_or_series_time_zone,
     get_expr_name,
     get_frequency_spectrum,
@@ -1558,6 +1559,22 @@ class TestFiniteEWMWeights:
             _ = _finite_ewm_weights(min_weight=1.0)
 
 
+class TestFirstTrueHorizontal:
+    @mark.parametrize(
+        ("row", "expected"),
+        [
+            param([True, True, True], 0),
+            param([False, True, True], 1),
+            param([False, False, True], 2),
+            param([False, False, False], None),
+        ],
+    )
+    def test_main(self, *, row: list[bool], expected: int | None) -> None:
+        df = DataFrame([row], schema={str(i): Boolean for i in range(3)}, orient="row")
+        result = first_true_horizontal(df)
+        assert result.item() == expected
+
+
 class TestGetDataTypeOrSeriesTimeZone:
     @given(
         time_zone=sampled_from([HongKong, UTC]),

{dycw_utilities-0.159.1 → dycw_utilities-0.159.2}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.159.1"
+__version__ = "0.159.2"

{dycw_utilities-0.159.1 → dycw_utilities-0.159.2}/src/utilities/polars.py

@@ -32,6 +32,7 @@ from polars import (
     any_horizontal,
     col,
     concat,
+    concat_list,
     datetime_range,
     int_range,
     lit,
@@ -1323,6 +1324,15 @@ class _FiniteEWMWeightsError(Exception):
 ##
 
 
+def first_true_horizontal(df: DataFrame, /) -> Series:
+    """Get the index of the first true in each row."""
+    expr = when(any_horizontal(pl.all())).then(concat_list(pl.all()).list.arg_max())
+    return one_column(df.select(expr))
+
+
+##
+
+
 def get_data_type_or_series_time_zone(
     dtype_or_series: PolarsDataType | Series, /
 ) -> ZoneInfo:
@@ -2583,6 +2593,7 @@ __all__ = [
     "ensure_expr_or_series_many",
     "expr_to_series",
     "finite_ewm_mean",
+    "first_true_horizontal",
     "get_data_type_or_series_time_zone",
     "get_expr_name",
     "get_frequency_spectrum",