如何用SymPy解析复杂多项式表达式并提取其关键信息？

核心思路 #

SymPy完全可以处理你提到的这类Mathematica格式多项式，关键是把解析后的Add类型表达式转换为Poly（多项式）对象，之后就能轻松提取各项信息。

分步实现代码示例 #

from sympy.parsing.mathematica import parse_mathematica
from sympy import symbols, Poly, sqrt, RootOf

# 定义变量
u = symbols('u')

# 1. 解析Mathematica格式的表达式字符串
expr1 = parse_mathematica("-1/4 + 1/(2*Sqrt[5]) + u^2")
expr2 = parse_mathematica("u^2 + Root[1 + 84*#1 + 560*#1^2 + 448*#1^3 & , 2, 0]")

# 2. 转换为Poly多项式对象（指定变量u）
poly1 = Poly(expr1, u)
poly2 = Poly(expr2, u)

# 3. 提取关键信息
# 针对第一个表达式
print("多项式1次数:", poly1.degree())
print("多项式1所有系数(从高次到常数项):", poly1.all_coeffs())
print("是否包含根号:", expr1.has(sqrt))

# 针对第二个含Root项的表达式
print("\n多项式2次数:", poly2.degree())
print("多项式2所有系数:", poly2.all_coeffs())
# 获取Root对应的极小多项式
root_term = expr2.args[1]
print("Root项对应的极小多项式:", root_term.args[0])

关键细节说明 #

• 解析表达式：parse_mathematica会将Mathematica语法的字符串转为SymPy的表达式对象（比如Add类型），这是后续处理的基础。
• 转Poly对象：Poly(expr, var)函数会把表达式按指定变量整理成标准多项式结构，自动处理项的合并与排序。
• 提取信息：
  • 次数：调用poly.degree()直接获取最高次幂
  • 系数：poly.all_coeffs()返回从最高次到常数项的完整系数列表；poly.coeffs()只返回非零系数
  • 根号判断：用expr.has(sqrt)检查表达式中是否存在根号类型的项
• 处理Root项：Mathematica的Root会被解析为SymPy的RootOf对象，这类对象属于代数常数（相对于变量u），会被当作多项式的常数项处理。若要查看该根对应的极小多项式，直接访问RootOf对象的args[0]即可。

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