如何在不转换为字符串的情况下将和拆分成几个部分

2022-04-06 00:00:00 python sympy

问题描述

假设我有一个表达式

from sympy import *
a,b,c,x,y = symbols('a c b x y') 
eq=a*x + b*x*y + c*y**2

需要拆分到包含单项的数组中。

我目前的解决方案是

parts = str(eq).split(' + ')

然后,我在要解释为表达式的数组部分的每个元素上使用val函数。

如何才能在不先将表达式转换为字符串的情况下将多元多项式拆分为单项式部分?


解决方案

您可以explore a sympy expression using .func and .args

eq.func
> <class 'sympy.core.add.Add'>
eq.args
> (a*x, b*y**2, c*x*y)

这些参数中的每一个都是sympy表达式,可以通过相同的方式进行研究:

eq.args[0].func
> <class 'sympy.core.mul.Mul'
eq.args[0].args
> (a, x)

等等。请注意,在表达式树的最后一级,您需要.func.args以外的其他函数,例如:

eq.args[0].args[0].name  # the a in a*x
> 'a'
eq.args[1].args[1].args[1].n()  # the 2 in y**2
> 2.00000000000000

相关文章