如何正确解析SymPy中含表达式的ITE条件语句字符串对象？

问题重现 #

正常运行场景 #

当解析单符号参数的ITE表达式时，代码可正常执行：

from sympy.parsing.sympy_parser import parse_expr
import sympy as sp

e = "ITE(1<2, K, X)"
e = parse_expr(e, evaluate=False)
print(f"e: {e}") # 输出 e: K

# 符号替换也正常
k = sp.Symbol("K")
e = e.subs(k, 1)
print(f"e: {e}") # 输出 e: 1

报错场景 #

当ITE的第二个参数为复合表达式（如K+K）时，解析触发错误：

e = "ITE(1<2, (K+K), X)"
e = parse_expr(e, evaluate=False)
print(f"e: {e}")

报错信息：

TypeError: expecting bool or Boolean, not K + K

问题根源：SymPy内置的ITE是布尔逻辑专用构造，要求分支返回布尔类型；而实际需求是支持任意符号/数值表达式的条件分支，这需要用SymPy的Piecewise，但默认parse_expr无法直接将字符串中的ITE映射为Piecewise。

无需修改输入字符串的解决方案 #

通过自定义parse_expr的转换规则，自动将字符串中的ITE语法转换为Piecewise表达式，具体代码如下：

from sympy.parsing.sympy_parser import parse_expr, standard_transformations, convert_xor
from sympy import Piecewise, Symbol
import re

# 自定义转换：将ITE(cond, expr1, expr2)转换为Piecewise((expr1, cond), (expr2, True))
def convert_ite(s, local_dict, global_dict):
    # 替换ITE开头为Piecewise的格式
    s = s.replace("ITE(", "Piecewise((")
    # 正则匹配并调整参数顺序与结构
    pattern = re.compile(r'Piecewise\(\((.*?),\s*(.*?)\),\s*(.*?)\)')
    def replace_match(match):
        expr1 = match.group(1)
        cond = match.group(2)
        expr2 = match.group(3)
        return f"Piecewise(({expr1}, {cond}), ({expr2}, True))"
    s = pattern.sub(replace_match, s)
    return s

# 组合转换规则：标准转换 + 自定义ITE转换
transformations = standard_transformations + (convert_xor, convert_ite)

# 测试解析复合表达式
e = "ITE(1<2, (K+K), X)"
expr = parse_expr(e, transformations=transformations, evaluate=False)
print(f"解析结果: {expr}") # 输出 Piecewise((K + K, 1 < 2), (X, True))

# 符号替换测试
k = Symbol("K")
expr_sub = expr.subs(k, 1)
print(f"替换后结果: {expr_sub}") # 输出 Piecewise((2, True), (X, True))

# 自动简化结果（设置evaluate=True）
expr_eval = parse_expr(e, transformations=transformations, evaluate=True)
print(f"自动简化结果: {expr_eval}") # 输出 2

说明 #

• 自定义convert_ite函数通过正则替换，将字符串中的ITE语法映射为SymPy标准的条件分支构造Piecewise，完全适配符号/数值表达式场景。
• 该方案无需修改原始输入字符串，直接通过解析器转换实现需求。
• 若需支持多层嵌套ITE，可调整正则表达式以处理复杂嵌套括号结构（当前正则已支持简单嵌套场景）。

内容的提问来源于stack exchange，提问作者nbes