polars-ta 0.5.1__tar.gz → 0.5.3__tar.gz

{polars_ta-0.5.1 → polars_ta-0.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polars_ta
-Version: 0.5.1
+Version: 0.5.3
 Summary: polars expressions
 Author-email: wukan <wu-kan@163.com>
 License: MIT License
@@ -31,7 +31,7 @@ Classifier: Programming Language :: Python
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: polars>=1.26.0
+Requires-Dist: polars>=1.28.0
 Requires-Dist: polars-ols>=0.3.0
 Requires-Dist: numpy
 Requires-Dist: numba

polars_ta-0.5.3/polars_ta/_version.py

@@ -0,0 +1 @@
+__version__ = "0.5.3"

{polars_ta-0.5.1 → polars_ta-0.5.3}/polars_ta/wq/_nb.py

@@ -181,18 +181,20 @@ def roll_triple_corr(x1, x2, x3, window, min_periods):
 
 @jit(nopython=True, nogil=True, fastmath=True, cache=True)
 def _cum_prod_by(r, by):
+    out = by.copy()
     for i in range(1, r.shape[0]):
-        if isnan(by[i]):
-            by[i] = r[i] * by[i - 1]
-    return by
+        if isnan(out[i]):
+            out[i] = r[i] * out[i - 1]
+    return out
 
 
 @jit(nopython=True, nogil=True, fastmath=True, cache=True)
 def _cum_sum_by(r, by):
+    out = by.copy()
     for i in range(1, r.shape[0]):
-        if isnan(by[i]):
-            by[i] = r[i] + by[i - 1]
-    return by
+        if isnan(out[i]):
+            out[i] = r[i] + out[i - 1]
+    return out
 
 
 @jit(nopython=True, nogil=True, fastmath=True, cache=True)

{polars_ta-0.5.1 → polars_ta-0.5.3}/polars_ta/wq/arithmetic.py

@@ -1,5 +1,6 @@
 import numpy as np
-from polars import Expr, Series, fold, any_horizontal, Float64
+from polars import Expr, Series, fold, any_horizontal, Float64, Int64
+from polars import arctan2 as _arctan2
 from polars import max_horizontal, sum_horizontal, min_horizontal, mean_horizontal
 
 
@@ -92,6 +93,16 @@ def arc_tan(x: Expr) -> Expr:
     return x.arctan()
 
 
+def arc_tan2(y: Expr, x: Expr) -> Expr:
+    """反正切二值函数"""
+    return _arctan2(y, x)
+
+
+def cbrt(x: Expr) -> Expr:
+    """立方根"""
+    return x.cbrt()
+
+
 def ceiling(x: Expr) -> Expr:
     """向上取整"""
     return x.ceil()
@@ -107,10 +118,57 @@ def cosh(x: Expr) -> Expr:
     return x.cosh()
 
 
+def cot(x: Expr) -> Expr:
+    """余切"""
+    return x.cot()
+
+
+def cube(x: Expr) -> Expr:
+    """立方"""
+    return x.pow(3)
+
+
+def degrees(x: Expr) -> Expr:
+    """弧度转角度"""
+    return x.degrees()
+
+
 def densify(x: Expr) -> Expr:
     raise
 
 
+def div(x: Expr, y: Expr) -> Expr:
+    """x/y的整数部分
+
+    Examples
+    --------
+    ```python
+    df = pl.DataFrame({
+        'a': [None, -1.5, 0., 1.5, 2.5],
+        'b': [None, -1, 0, 1, 2],
+    }).with_columns(
+        out1=div(pl.col('a'), 0),
+        out2=div(pl.col('a'), 1),
+        out3=div(pl.col('a'), pl.col('b')),
+    )
+    shape: (5, 5)
+    ┌──────┬──────┬──────┬──────┬──────┐
+    │ a    ┆ b    ┆ out1 ┆ out2 ┆ out3 │
+    │ ---  ┆ ---  ┆ ---  ┆ ---  ┆ ---  │
+    │ f64  ┆ i64  ┆ i64  ┆ i64  ┆ i64  │
+    ╞══════╪══════╪══════╪══════╪══════╡
+    │ null ┆ null ┆ null ┆ null ┆ null │
+    │ -1.5 ┆ -1   ┆ null ┆ -2   ┆ 1    │
+    │ 0.0  ┆ 0    ┆ null ┆ 0    ┆ null │
+    │ 1.5  ┆ 1    ┆ null ┆ 1    ┆ 1    │
+    │ 2.5  ┆ 2    ┆ null ┆ 2    ┆ 1    │
+    └──────┴──────┴──────┴──────┴──────┘
+    ```
+
+    """
+    return x.floordiv(y).cast(Int64, strict=False)
+
+
 def divide(x: Expr, y: Expr) -> Expr:
     """x/y
 
@@ -368,6 +426,35 @@ def log1p(x: Expr) -> Expr:
     return x.log1p()
 
 
+def log2(x: Expr) -> Expr:
+    """2为底的对数
+
+    Examples
+    --------
+    ```python
+    df = pl.DataFrame({
+        'a': [None, -1, 0, 1, 2],
+    }).with_columns(
+        out1=log2(pl.col('a')),
+    )
+    shape: (5, 2)
+    ┌──────┬──────┐
+    │ a    ┆ out1 │
+    │ ---  ┆ ---  │
+    │ i64  ┆ f64  │
+    ╞══════╪══════╡
+    │ null ┆ null │
+    │ -1   ┆ NaN  │
+    │ 0    ┆ -inf │
+    │ 1    ┆ 0.0  │
+    │ 2    ┆ 1.0  │
+    └──────┴──────┘
+    ```
+
+    """
+    return x.log(2)
+
+
 def max_(a: Expr, b: Expr, *args) -> Expr:
     """水平多列最大值 Maximum value of all inputs. At least 2 inputs are required."""
     return max_horizontal(a, b, *args)
@@ -515,6 +602,11 @@ def power(x: Expr, y: Expr) -> Expr:
     return x.pow(y.cast(Float64))
 
 
+def radians(x: Expr) -> Expr:
+    """角度转弧度"""
+    return x.radians()
+
+
 def reverse(x: Expr) -> Expr:
     """-x"""
     return -x
@@ -735,6 +827,11 @@ def sqrt(x: Expr) -> Expr:
     return x.sqrt()
 
 
+def square(x: Expr) -> Expr:
+    """平方"""
+    return x.pow(2)
+
+
 def subtract(x: Expr, y: Expr) -> Expr:
     """x-y"""
     return x - y

{polars_ta-0.5.1 → polars_ta-0.5.3}/polars_ta/wq/time_series.py

@@ -7,7 +7,7 @@ from polars_ols import RollingKwargs
 
 import polars_ta
 from polars_ta.utils.numba_ import batches_i1_o1, batches_i2_o1, batches_i2_o2, struct_to_numpy
-from polars_ta.utils.pandas_ import roll_kurt, roll_rank
+from polars_ta.utils.pandas_ import roll_rank
 from polars_ta.wq._nb import roll_argmax, roll_argmin, roll_co_kurtosis, roll_co_skewness, roll_moment, roll_partial_corr, roll_triple_corr, _cum_prod_by, _cum_sum_by, _signals_to_size, _cum_sum_reset, _sum_split_by, roll_decay_linear, roll_decay_exp_window, roll_prod
 
 
@@ -555,18 +555,51 @@ def ts_ir(x: Expr, d: int = 1, min_samples: Optional[int] = None) -> Expr:
     return ts_mean(x, d, min_samples) / ts_std_dev(x, d, 0, min_samples)
 
 
-def ts_kurtosis(x: Expr, d: int = 5, min_samples: Optional[int] = None) -> Expr:
+def ts_kurtosis(x: Expr, d: int = 5, bias: bool = False, min_samples: Optional[int] = None) -> Expr:
     """kurtosis of x for the last d days
 
     时序滚动峰度
 
-    Warnings
+    Parameters
+    ----------
+    x
+    d
+    bias
+        有偏
+    min_samples
+
+    Notes
+    -----
+    `bias=False`时与`pandas`结果一样
+
+    Examples
     --------
-    等待polars官方出rolling_kurt
+    ```python
+    df = pl.DataFrame({
+        'a': [None, 1, 2, 3, 4, 999],
+    }).with_columns(
+        out1=pl.col('a').map_batches(lambda x: pl.Series(pd.Series(x).rolling(4).kurt())),
+        out2=ts_kurtosis(pl.col('a'), 4),
+    )
+
+    shape: (6, 3)
+    ┌──────┬──────────┬──────────┐
+    │ a    ┆ out1     ┆ out2     │
+    │ ---  ┆ ---      ┆ ---      │
+    │ i64  ┆ f64      ┆ f64      │
+    ╞══════╪══════════╪══════════╡
+    │ null ┆ null     ┆ null     │
+    │ 1    ┆ null     ┆ null     │
+    │ 2    ┆ null     ┆ null     │
+    │ 3    ┆ null     ┆ null     │
+    │ 4    ┆ -1.2     ┆ -1.2     │
+    │ 999  ┆ 3.999946 ┆ 3.999946 │
+    └──────┴──────────┴──────────┘
+    ```
 
     """
-    minp = min_samples or polars_ta.MIN_SAMPLES or d
-    return x.map_batches(lambda a: roll_kurt(a, d, minp))
+    minp = min_samples or polars_ta.MIN_SAMPLES
+    return x.rolling_kurtosis(d, min_samples=minp, bias=bias)
 
 
 def ts_l2_norm(x: Expr, d: int = 5, min_samples: Optional[int] = None) -> Expr:
@@ -705,7 +738,7 @@ def ts_scale(x: Expr, d: int = 5, min_samples: Optional[int] = None) -> Expr:
     return when(a != b).then((x - a) / (b - a)).otherwise(0)
 
 
-def ts_skewness(x: Expr, d: int = 5, bias: bool = False) -> Expr:
+def ts_skewness(x: Expr, d: int = 5, bias: bool = False, min_samples: Optional[int] = None) -> Expr:
     """Return skewness of x for the past d days
 
     时序滚动偏度
@@ -716,13 +749,15 @@ def ts_skewness(x: Expr, d: int = 5, bias: bool = False) -> Expr:
     d
     bias
         有偏
+    min_samples
 
     Notes
     -----
     `bias=False`时与`pandas`结果一样
 
     """
-    return x.rolling_skew(d, bias=bias)
+    minp = min_samples or polars_ta.MIN_SAMPLES
+    return x.rolling_skew(d, min_samples=minp, bias=bias)
 
 
 def ts_std_dev(x: Expr, d: int = 5, ddof: int = 0, min_samples: Optional[int] = None) -> Expr:

{polars_ta-0.5.1 → polars_ta-0.5.3}/polars_ta.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polars_ta
-Version: 0.5.1
+Version: 0.5.3
 Summary: polars expressions
 Author-email: wukan <wu-kan@163.com>
 License: MIT License
@@ -31,7 +31,7 @@ Classifier: Programming Language :: Python
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: polars>=1.26.0
+Requires-Dist: polars>=1.28.0
 Requires-Dist: polars-ols>=0.3.0
 Requires-Dist: numpy
 Requires-Dist: numba

{polars_ta-0.5.1 → polars_ta-0.5.3}/polars_ta.egg-info/requires.txt

@@ -1,4 +1,4 @@
-polars>=1.26.0
+polars>=1.28.0
 polars-ols>=0.3.0
 numpy
 numba

{polars_ta-0.5.1 → polars_ta-0.5.3}/pyproject.toml

@@ -17,7 +17,7 @@ classifiers = [
     "Programming Language :: Python"
 ]
 dependencies = [
-    "polars>=1.26.0",
+    "polars>=1.28.0",
     "polars-ols>=0.3.0",
     "numpy",
     "numba",

polars_ta-0.5.1/polars_ta/_version.py

@@ -1 +0,0 @@
-__version__ = "0.5.1"