SymPy中判断表达式是否为三角函数及字典遍历问题

1. 无手动枚举判断三角函数 #

利用SymPy内置的三角函数基类TrigonometricFunction，可以直接判断表达式是否属于三角函数范畴，无需手动枚举每个三角函数：

from sympy import sin, cos, symbols
from sympy.functions.elementary.trigonometric import TrigonometricFunction

x = symbols('x')
trig_expr = sin(x)
non_trig_expr = x**2

# 判断是否为三角函数
print(isinstance(trig_expr, TrigonometricFunction))  # 输出 True
print(isinstance(non_trig_expr, TrigonometricFunction))  # 输出 False

该基类覆盖了所有标准三角函数（sin、cos、tan、csc、sec、cot等），完全满足无枚举判断的需求。

2. SymPy表达式的分类可行性 #

完全可以对SymPy表达式进行分类，有两种常用方式：

• 利用SymPy内置属性：SymPy为表达式提供了多个内置布尔属性，可快速判定类型：

  from sympy import sin, exp, symbols
  x = symbols('x')
  expr = sin(x) + exp(x)
  
  print(expr.is_trig)  # 判定是否为三角函数，输出 True
  print(expr.is_exp)   # 判定是否为指数函数，输出 True
  print(expr.is_poly)  # 判定是否为多项式，输出 False
  

  其他常用属性还包括is_rational（有理式）、is_log（对数函数）等。
• 自定义分类逻辑：如果内置属性无法满足需求，可通过检查表达式的类、结构或操作符实现自定义分类：

  from sympy import symbols, sin, exp, Poly
  from sympy.functions.elementary.trigonometric import TrigonometricFunction
  x = symbols('x')
  
  def classify_expression(expr):
      if isinstance(expr, TrigonometricFunction):
          return "三角函数"
      elif expr.is_exp:
          return "指数函数"
      elif isinstance(expr, Poly):
          return "多项式"
      else:
          return "其他类型"
  
  print(classify_expression(sin(x)))  # 输出 三角函数
  print(classify_expression(exp(x)))  # 输出 指数函数
  print(classify_expression(x**3))    # 输出 多项式
  

3. 遍历字典获取元组元素并触发警告 #

遍历字典时，直接对键元组进行解构即可获取参数数量；结合三角函数判断逻辑，可在参数数量不符合要求时抛出警告：

import warnings
from sympy import sin, cos, symbols
from sympy.functions.elementary.trigonometric import TrigonometricFunction

x, y = symbols('x y')
# 示例字典
transform_templates = {
    (sin(x), 1): "单参数三角函数模板",
    (cos(x+y), 2): "双参数三角函数模板",
    (x**2, 1): "多项式模板"
}

# 遍历字典并解构键元组
for (expr, param_count), template in transform_templates.items():
    print(f"表达式: {expr}, 参数数量: {param_count}, 对应模板: {template}")
    # 检查三角函数参数数量是否合规
    if isinstance(expr, TrigonometricFunction) and param_count != 1:
        warnings.warn(f"警告：三角函数 {expr} 的参数数量应为1，当前为{param_count}", UserWarning)

运行后会针对cos(x+y)触发警告，符合需求。

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