dycw-utilities 0.109.21__py3-none-any.whl → 0.109.23__py3-none-any.whl

{dycw_utilities-0.109.21.dist-info → dycw_utilities-0.109.23.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.109.21
+Version: 0.109.23
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
 Requires-Dist: typing-extensions<4.14,>=4.13.1
 Provides-Extra: test
-Requires-Dist: hypothesis<6.132,>=6.131.7; extra == 'test'
+Requires-Dist: hypothesis<6.132,>=6.131.9; extra == 'test'
 Requires-Dist: pytest-asyncio<0.27,>=0.26.0; extra == 'test'
 Requires-Dist: pytest-cov<6.2,>=6.1.1; extra == 'test'
 Requires-Dist: pytest-instafail<0.6,>=0.5.0; extra == 'test'
@@ -80,7 +80,7 @@ Provides-Extra: zzz-test-hypothesis
 Requires-Dist: aiosqlite<0.22,>=0.21.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: asyncpg<0.31,>=0.30.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: greenlet<3.3,>=3.2.0; extra == 'zzz-test-hypothesis'
-Requires-Dist: hypothesis<6.132,>=6.131.7; extra == 'zzz-test-hypothesis'
+Requires-Dist: hypothesis<6.132,>=6.131.9; extra == 'zzz-test-hypothesis'
 Requires-Dist: luigi<3.7,>=3.6.0; extra == 'zzz-test-hypothesis'
 Requires-Dist: numpy<2.3,>=2.2.5; extra == 'zzz-test-hypothesis'
 Requires-Dist: pathvalidate<3.3,>=3.2.3; extra == 'zzz-test-hypothesis'

{dycw_utilities-0.109.21.dist-info → dycw_utilities-0.109.23.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=Q2YV4YDdpMRcnIlWhuT_v301enhQDMFopz9BOGv-13g,61
+utilities/__init__.py,sha256=4lEn-RM4k4cprCQpv11-dCWVShqTEywTV3oeqqUTdtc,61
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/astor.py,sha256=xuDUkjq0-b6fhtwjhbnebzbqQZAjMSHR1IIS5uOodVg,777
 utilities/asyncio.py,sha256=41oQUurWMvadFK5gFnaG21hMM0Vmfn2WS6OpC0R9mas,14757
@@ -46,7 +46,7 @@ utilities/pathlib.py,sha256=31WPMXdLIyXgYOMMl_HOI2wlo66MGSE-cgeelk-Lias,1410
 utilities/period.py,sha256=ikHXsWtDLr553cfH6p9mMaiCnIAP69B7q84ckWV3HaA,10884
 utilities/pickle.py,sha256=Bhvd7cZl-zQKQDFjUerqGuSKlHvnW1K2QXeU5UZibtg,657
 utilities/platform.py,sha256=NU7ycTvAXAG-fdYmDXaM1m4EOml2cGiaYwaUzfzSqyU,1767
-utilities/polars.py,sha256=27oT-CmOtlAEdsZ2XOoN1GLINnRM2Byj_oBtsHGmp3c,57444
+utilities/polars.py,sha256=woTGhyzXNLN30SQgwXz54-I1aJp1oWATJ2rpmke7gKI,58419
 utilities/polars_ols.py,sha256=efhXf0gjrHUpQrvS6a7g8yJQJWf_ATKtJnqqF2inCOU,5680
 utilities/pqdm.py,sha256=foRytQybmOQ05pjt5LF7ANyzrIa--4ScDE3T2wd31a4,3118
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -68,7 +68,7 @@ utilities/slack_sdk.py,sha256=SeDNMh24IPiEBWoGMdgvrflUaFa9TGlTS03H9-NKaQw,4132
 utilities/socket.py,sha256=K77vfREvzoVTrpYKo6MZakol0EYu2q1sWJnnZqL0So0,118
 utilities/sqlalchemy.py,sha256=GWzp54TP3F2mGhxPTn0c56KxxDeN9VKLMagcRSELhf4,35453
 utilities/sqlalchemy_polars.py,sha256=oGyMX5gSxuLI3N8mtz_-ml3UdWKcZuj6aFRW6ifI0Kc,15617
-utilities/statsmodels.py,sha256=F0bNmPUCmi3GsfpSuTVBYWxGimKury-pmcpFQ-SJ65I,2629
+utilities/statsmodels.py,sha256=koyiBHvpMcSiBfh99wFUfSggLNx7cuAw3rwyfAhoKpQ,3410
 utilities/streamlit.py,sha256=U9PJBaKP1IdSykKhPZhIzSPTZsmLsnwbEPZWzNhJPKk,2955
 utilities/sys.py,sha256=h0Xr7Vj86wNalvwJVP1wj5Y0kD_VWm1vzuXZ_jw94mE,2743
 utilities/tempfile.py,sha256=VqmZJAhTJ1OaVywFzk5eqROV8iJbW9XQ_QYAV0bpdRo,1384
@@ -87,7 +87,7 @@ utilities/warnings.py,sha256=yUgjnmkCRf6QhdyAXzl7u0qQFejhQG3PrjoSwxpbHrs,1819
 utilities/whenever.py,sha256=TjoTAJ1R27-rKXiXzdE4GzPidmYqm0W58XydDXp-QZM,17786
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-DQz5a0Ikw9jfSZtL0BEQkXOMC9yGn_xiJYNCLMiqEc,1989
-dycw_utilities-0.109.21.dist-info/METADATA,sha256=oeS1xA25IzXCADvmuZu2P1ULQ3p1Kyv0_tkEcZmZzDo,13005
-dycw_utilities-0.109.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.109.21.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.109.21.dist-info/RECORD,,
+dycw_utilities-0.109.23.dist-info/METADATA,sha256=fMol_EYcAIARY4nGcAjYEe1AXs-5dAorn3vGy27XdBY,13005
+dycw_utilities-0.109.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.109.23.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.109.23.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.109.21"
+__version__ = "0.109.23"

utilities/polars.py

@@ -140,6 +140,32 @@ DatetimeUSEastern = Datetime(time_zone="US/Eastern")
 DatetimeUTC = Datetime(time_zone="UTC")
 _FINITE_EWM_MIN_WEIGHT = 0.9999
 
+##
+
+
+def ac_halflife(
+    series: Series,
+    /,
+    *,
+    adjusted: bool = False,
+    fft: bool = True,
+    bartlett_confint: bool = True,
+    missing: ACFMissing = "none",
+    step: float = 0.01,
+) -> float:
+    """Compute the autocorrelation halflife."""
+    import utilities.statsmodels
+
+    array = series.to_numpy()
+    return utilities.statsmodels.ac_halflife(
+        array,
+        adjusted=adjusted,
+        fft=fft,
+        bartlett_confint=bartlett_confint,
+        missing=missing,
+        step=step,
+    )
+
 
 ##
 
@@ -223,6 +249,11 @@ def _acf_process_qstats_pvalues(qstats: NDArrayF, pvalues: NDArrayF, /) -> DataF
 ##
 
 
+# def acf_halflife(series: Series,/)
+
+##
+
+
 def adjust_frequencies(
     series: Series,
     /,
@@ -1197,6 +1228,21 @@ class _GetDataTypeOrSeriesTimeZoneNotZonedError(GetDataTypeOrSeriesTimeZoneError
 ##
 
 
+def get_expr_name(obj: Series | DataFrame, expr: Expr, /) -> str:
+    """Get the name of an expression."""
+    match obj:
+        case Series() as series:
+            return get_expr_name(series.to_frame(), expr)
+        case DataFrame() as df:
+            selected = df.select(expr)
+            return one(selected.columns)
+        case _ as never:
+            assert_never(never)
+
+
+##
+
+
 def get_frequency_spectrum(series: Series, /, *, d: int = 1) -> DataFrame:
     """Get the frequency spectrum."""
     import utilities.numpy
@@ -1939,6 +1985,7 @@ __all__ = [
     "finite_ewm_mean",
     "floor_datetime",
     "get_data_type_or_series_time_zone",
+    "get_expr_name",
     "get_frequency_spectrum",
     "get_series_number_of_decimals",
     "insert_after",

utilities/statsmodels.py

@@ -2,12 +2,45 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, Literal, cast, overload
 
+import numpy as np
 import statsmodels.tsa.stattools
+from numpy import arange, argmax, interp, nan
+
+from utilities.numpy import shift
 
 if TYPE_CHECKING:
     from utilities.numpy import NDArrayF
 
 
+def ac_halflife(
+    array: NDArrayF,
+    /,
+    *,
+    adjusted: bool = False,
+    fft: bool = True,
+    bartlett_confint: bool = True,
+    missing: ACFMissing = "none",
+    step: float = 0.01,
+) -> float:
+    """Compute the autocorrelation halflife."""
+    (n,) = array.shape
+    acfs = acf(
+        array,
+        adjusted=adjusted,
+        nlags=n,
+        fft=fft,
+        bartlett_confint=bartlett_confint,
+        missing=missing,
+    )
+    lags = arange(0, n, step=step)
+    interp_acfs = interp(lags, arange(n), acfs)
+    is_half = (shift(interp_acfs) > 0.5) & (interp_acfs <= 0.5)
+    return lags[argmax(is_half)].item() if np.any(is_half) else nan
+
+
+##
+
+
 type ACFMissing = Literal["none", "raise", "conservative", "drop"]
 
 
@@ -114,4 +147,4 @@ def acf(
     )
 
 
-__all__ = ["ACFMissing", "acf"]
+__all__ = ["ACFMissing", "ac_halflife", "acf"]