polars-ta 0.5.2__tar.gz → 0.5.4__tar.gz

polars_ta-0.5.4/.gitignore

@@ -0,0 +1,160 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/

{polars_ta-0.5.2 → polars_ta-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polars_ta
-Version: 0.5.2
+Version: 0.5.4
 Summary: polars expressions
 Author-email: wukan <wu-kan@163.com>
 License: MIT License
@@ -24,21 +24,19 @@ License: MIT License
         LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
         OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
         SOFTWARE.
-        
-Keywords: polars,expression,talib
+License-File: LICENSE
+Keywords: expression,polars,talib
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
 Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: polars>=1.28.0
-Requires-Dist: polars-ols>=0.3.0
-Requires-Dist: numpy
 Requires-Dist: numba
+Requires-Dist: numpy
 Requires-Dist: pandas
+Requires-Dist: polars-ols>=0.3.0
+Requires-Dist: polars>=1.28.0
 Provides-Extra: talib
-Requires-Dist: TA-Lib; extra == "talib"
-Dynamic: license-file
+Requires-Dist: ta-lib; extra == 'talib'
+Description-Content-Type: text/markdown
 
 # polars_ta
 
@@ -248,6 +246,11 @@ mkdocs build
 也可以通过以下链接导入：
 https://polars-ta.readthedocs.io/en/latest/llms-full.txt
 
+## 提示词
+由于`llms-full.txt`信息不适合做提示词，所以`tools/prompt.py`提供了生成更简洁算子清单的功能。
+
+用户也可以直接使用`prompt.txt`(欢迎提示词工程专家帮忙改进，做的更准确)
+
 ## 参考
 
 - https://github.com/pola-rs/polars

{polars_ta-0.5.2 → polars_ta-0.5.4}/README.md

@@ -206,6 +206,11 @@ mkdocs build
 也可以通过以下链接导入：
 https://polars-ta.readthedocs.io/en/latest/llms-full.txt
 
+## 提示词
+由于`llms-full.txt`信息不适合做提示词，所以`tools/prompt.py`提供了生成更简洁算子清单的功能。
+
+用户也可以直接使用`prompt.txt`(欢迎提示词工程专家帮忙改进，做的更准确)
+
 ## 参考
 
 - https://github.com/pola-rs/polars

polars_ta-0.5.4/polars_ta/_version.py

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

polars_ta-0.5.4/polars_ta/labels/__init__.py

@@ -0,0 +1 @@
+from polars_ta.labels.future import *  # noqa

polars_ta-0.5.4/polars_ta/labels/_nb.py

@@ -0,0 +1,40 @@
+import numpy as np
+from numba import jit
+from numpy import full
+
+
+@jit(nopython=True, nogil=True, cache=True)
+def _triple_barrier(close: np.ndarray, high: np.ndarray, low: np.ndarray, window: int, take_profit: float, stop_loss: float) -> np.ndarray:
+    """三重障碍打标法"""
+    out = full(close.shape[0], np.nan, dtype=np.float64)
+    for i in range(close.shape[0] - window + 1):
+        entry_price = close[i]
+        if np.isnan(entry_price):
+            # out[i] = 0
+            continue
+        upper_barrier = entry_price * (1 + take_profit)
+        lower_barrier = entry_price * (1 - stop_loss)
+        for j in range(i + 1, i + window):
+            hit_upper = high[j] >= upper_barrier
+            hit_lower = low[j] <= lower_barrier
+            if hit_upper and hit_lower:
+                # TODO 同一天无法知道是先触发止损还是先触发止盈
+                # 1. 假定离收盘价远的先触发
+                if high[j] - close[j] > close[j] - low[j]:
+                    out[i] = 1  # 最高价更远，触发止盈
+                else:
+                    out[i] = -1  # 最低价更远，触发止损
+
+                # out[i] = -1 # 2. 简化处理认为先触发止损
+                break
+            if hit_upper:
+                out[i] = 1  # 止盈
+                break
+            if hit_lower:
+                out[i] = -1  # 止损
+                break
+        else:
+            # out[i] = 0  # 1. 时间到了触发平仓
+            out[i] = np.sign(close[j] / entry_price - 1)  # 2. 时间到了触发平仓
+
+    return out

polars_ta-0.5.4/polars_ta/labels/future.py

@@ -0,0 +1,165 @@
+"""
+
+由于标签的定义比较灵活，所以以下代码主要用于参考
+
+Notes
+-----
+标签都是未来数据,在机器学习中，只能用于`y`,不能用于`X`。
+
+References
+----------
+https://mp.weixin.qq.com/s/XtgYezFsslOfW-QyIMr0VA
+https://github.com/Rachnog/Advanced-Deep-Trading/blob/master/bars-labels-diff/Labeling.ipynb
+
+"""
+from polars import Expr, struct
+
+from polars_ta.labels._nb import _triple_barrier
+from polars_ta.utils.numba_ import batches_i2_o1, struct_to_numpy
+from polars_ta.wq import cut, ts_delay, ts_log_diff, log
+
+
+def ts_log_return(close: Expr, n: int = 5) -> Expr:
+    """将未来数据当成卖出价后移到买入价位置，计算对数收益率
+
+    Examples
+    --------
+    ```python
+    df = pl.DataFrame({
+        'a': [None, 10, 11, 12, 9, 12, 13],
+    }).with_columns(
+        out1=ts_log_return(pl.col('a'), 3),
+        out2=_ts_log_return(pl.col('a'), 3),
+    )
+
+    shape: (7, 3)
+    ┌──────┬───────────┬───────────┐
+    │ a    ┆ out1      ┆ out2      │
+    │ ---  ┆ ---       ┆ ---       │
+    │ i64  ┆ f64       ┆ f64       │
+    ╞══════╪═══════════╪═══════════╡
+    │ null ┆ null      ┆ null      │
+    │ 10   ┆ -0.105361 ┆ -0.105361 │
+    │ 11   ┆ 0.087011  ┆ 0.087011  │
+    │ 12   ┆ 0.080043  ┆ 0.080043  │
+    │ 9    ┆ null      ┆ null      │
+    │ 12   ┆ null      ┆ null      │
+    │ 13   ┆ null      ┆ null      │
+    └──────┴───────────┴───────────┘
+    ```
+
+    """
+    # return (close.shift(-n) / close).log()
+    return log(ts_delay(close, -n) / close)
+
+
+def _ts_log_return(close: Expr, n: int = 5) -> Expr:
+    """计算对数收益率，但将结果后移
+
+    如果打标签方式复杂，这种最终结果后移的方法更方便
+    """
+    # return (close / close.shift(n)).log().shift(-n)
+    return ts_delay(ts_log_diff(close, n), -n)
+
+
+def ts_simple_return(close: Expr, n: int = 5, threshold: float = 0.0, *more_threshold) -> Expr:
+    """简单收益率标签。支持二分类、三分类等。对收益率使用`cut`进行分类
+    
+    Parameters
+    ----------
+    close
+    n:int
+        未来n天
+    threshold:float
+        收益率阈值，小于该值为0，大于等于该值为1
+    more_threshold:float
+        更多的阈值，用于三分类等。小于该值为1，大于等于该值为2，以此类推
+
+    Returns
+    -------
+    Expr
+        标签列, 类型为UInt32, 取值为0, 1, 2, ...
+
+    Examples
+    --------
+    ```python
+    df = pl.DataFrame({
+        'a': [None, 10., 9.99, 9., 10., 11., 11.],
+    }).with_columns(
+        out1=label_simple_return(pl.col('a'), 1, 0),
+        out2=label_simple_return(pl.col('a'), 1, -0.001, 0.001),
+    )
+
+    shape: (7, 3)
+    ┌──────┬──────┬──────┐
+    │ a    ┆ out1 ┆ out2 │
+    │ ---  ┆ ---  ┆ ---  │
+    │ f64  ┆ u32  ┆ u32  │
+    ╞══════╪══════╪══════╡
+    │ null ┆ null ┆ null │
+    │ 10.0 ┆ 0    ┆ 0    │
+    │ 9.99 ┆ 0    ┆ 0    │
+    │ 9.0  ┆ 1    ┆ 2    │
+    │ 10.0 ┆ 1    ┆ 2    │
+    │ 11.0 ┆ 0    ┆ 1    │
+    │ 11.0 ┆ null ┆ null │
+    └──────┴──────┴──────┘
+
+    """
+    return cut(close.pct_change(n).shift(-n), threshold, *more_threshold)
+
+
+def ts_triple_barrier(close: Expr, high: Expr, low: Expr, d: int = 5, take_profit: float = 0.1, stop_loss: float = 0.05) -> Expr:
+    """三重障碍打标法
+
+    Parameters
+    ----------
+    close:Expr
+        收盘价
+    high:Expr
+        最高价
+    low:Expr
+        最低价
+    d:int
+        时间窗口
+    take_profit:float
+        止盈比例
+    stop_loss:float
+        止损比例
+
+    Returns
+    -------
+    Expr
+        标签列。取值为-1止损, 1止盈，0时间到期
+
+    Notes
+    -----
+    1. `high`, `low`在粗略情况下可用`close`代替
+    2. 时间到期时，根据盈亏返回不同的标签
+
+    Examples
+    --------
+    ```python
+    df = pl.DataFrame({
+        "close": [np.nan, 1, 1, 1.0],
+        "high": [np.nan, 1, 1.1, 1],
+        "low": [np.nan, 1, 1, 0.95],
+    }).with_columns(
+        out=ts_triple_barrier(pl.col("close"), pl.col("high"), pl.col("low"), 2, 0.1, 0.05)
+    )
+
+    shape: (4, 4)
+    ┌───────┬──────┬──────┬──────┐
+    │ close ┆ high ┆ low  ┆ out  │
+    │ ---   ┆ ---  ┆ ---  ┆ ---  │
+    │ f64   ┆ f64  ┆ f64  ┆ f64  │
+    ╞═══════╪══════╪══════╪══════╡
+    │ NaN   ┆ NaN  ┆ NaN  ┆ null │
+    │ 1.0   ┆ 1.0  ┆ 1.0  ┆ 1.0  │
+    │ 1.0   ┆ 1.1  ┆ 1.0  ┆ -1.0 │
+    │ 1.0   ┆ 1.0  ┆ 0.95 ┆ null │
+    └───────┴──────┴──────┴──────┘
+    ```
+
+    """
+    return struct([close, high, low]).map_batches(lambda xx: batches_i2_o1(struct_to_numpy(xx, 3), _triple_barrier, d, take_profit, stop_loss))

{polars_ta-0.5.2 → polars_ta-0.5.4}/polars_ta/performance/returns.py

@@ -1,12 +1,6 @@
 from polars import Expr
 
-from polars_ta.wq.arithmetic import log1p, expm1  # noqa
-# log return
-# 对数收益
-from polars_ta.wq.time_series import ts_log_diff as ts_log_return  # noqa
-# simple percentage return
-# 简单收益
-from polars_ta.wq.time_series import ts_returns as ts_percent_return  # noqa
+from polars_ta.wq.arithmetic import log1p, expm1
 
 
 def ts_cum_return(close: Expr) -> Expr:

polars_ta-0.5.4/polars_ta/prefix/labels.py

@@ -0,0 +1 @@
+from polars_ta.labels import *  # noqa

polars_ta-0.5.4/polars_ta/ta/README.md

@@ -0,0 +1,12 @@
+# polars_ta.ta
+
+1. Files in this folder mimic `talib`, and implement `polars` versions for the same functions
+2. Since we reduce the functino calls between `Python` and `C` code, it should be faster than `talib`.
+3. We first try to import from `ta`, then from `wq`, and only implement the function if it is not available.
+4. When there is a circular dependency, we use `polars` instead.
+
+
+1. 本文件夹中模仿`talib`，实现同名函数的`polars`版
+2. 由于减少了python与c来回调用，理论上比直接调用`talib`快
+3. 优先从`ta`中导入，然后从`wq`中导入，没有的才实现
+4. 出现循环依赖时，使用`polars`

polars_ta-0.5.4/polars_ta/talib/README.md

@@ -0,0 +1,12 @@
+# polars_ta.talib
+
+Inside this package, files are generated by `tools.codegen_talib2`.
+It is a wrapper of `talib` functions, with the following features:
+
+1. Input and output are `Expr` instead of `Series`
+2. ~~Add skipna feature (not efficient, will update when `polars` support backward fill)~~
+
+本包由`tools.codegen_talib2`自动生成，是对`talib`代码的封装。实现了以下功能
+1. 输入输出由`Series`改`Expr`
+2. ~~添加跳过空值功能（效率不高，等`polars`支持反向填充，此部分将更新）~~
+

polars_ta-0.5.4/polars_ta/tdx/README.md

@@ -0,0 +1,10 @@
+# polars_ta.tdx
+
+1. Follows the `tdx` naming convention
+2. Except for some element-wise functions, all functions are time-series functions. Pay special attention to `MAX` and similar functions.
+3. First import from `tdx`, then from `wq`, and finally from `ta`. Only implement the function if it is not available.
+
+
+1. 函数名称按照通达信来
+2. 除了部分按元素计算的函数，其它都为时序函数，特别注意`MAX`等一类不要混淆
+3. 优先从`tdx`中导入，然后从`wq`中导入，最后从`ta`，没有的才实现

{polars_ta-0.5.2 → polars_ta-0.5.4}/polars_ta/utils/numba_.py

@@ -27,6 +27,7 @@ def isnan(x):
 
 @jit(nopython=True, nogil=True, cache=True)
 def full_with_window_size(arr, fill_value, dtype=None, window_size: int = 1):
+    """创建一个更大的数组，填充后一截数据"""
     out = full(arr.shape[0] + window_size - 1, fill_value, dtype=dtype)
     out[window_size - 1:] = arr
     return out
@@ -34,6 +35,7 @@ def full_with_window_size(arr, fill_value, dtype=None, window_size: int = 1):
 
 @jit(nopython=True, nogil=True, cache=True)
 def sliding_window_with_min_periods(arr, window_size: int, min_periods: int):
+    """为rolling准备的数据，当数据长度不足时，用nan填充"""
     windows = sliding_window_view(arr, window_size)
     valid_counts = np.sum(~np.isnan(windows), axis=1)
     # 修改这一行，使用布尔索引而不是np.where