添加xor函数

wukan1986 · Nov 6, 2024 · 7f42e0b · 7f42e0b
1 parent d32332f
commit 7f42e0b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/polars_ta/_version.py b/polars_ta/_version.py
@@ -1 +1 @@
-__version__ = "0.2.11"
+__version__ = "0.2.12"
diff --git a/polars_ta/wq/logical.py b/polars_ta/wq/logical.py
@@ -2,8 +2,6 @@
 from polars import when, Int8, Boolean
 
 
-# TODO 本文件返回bool时是否有必要转换成 0/1 ?
-
 def and_(a: Expr, b: Expr, *args) -> Expr:
     """Logical AND operator, returns true if both operands are true and returns false otherwise"""
     return all_horizontal(a, b, *args)
@@ -54,6 +52,11 @@ def or_(a: Expr, b: Expr, *args) -> Expr:
     return any_horizontal(a, b, *args)
 
 
+def xor(a: Expr, b: Expr) -> Expr:
+    """Logical XOR operator returns true if exactly one of the inputs is true and returns false otherwise"""
+    return a.xor(b)
+
+
 def int_(a: Expr) -> Expr:
     """convert bool to int"""
     return a.cast(Int8)