Question

我目前正在处理多个变量的函数，需要收集类似的术语以尝试简化表达式。

假设表达式如下：

x = sympy.Symbol('x')
y = sympy.Symbol('y')
k = sympy.Symbol('k')
a = sympy.Symbol('a')

z = k*(y**2*(a + x) + (a + x)**3/3) - k((2*k*y*(a + x)*(n - 1)*(-k*(y**2*(-a + x) + (-a + x)**3/3) + k*(y**2*(a + x) + (a + x)**3/3)) + y)**2*(-a + k*(n - 1)*(y**2 + (a + x)**2)*(-k*(y**2*(-a + x)))))
zEx = z.expand()
print type(z)
print type(zEx)

编辑：格式化以增加清晰度并更改表达式z以使问题更易于理解。

说z包含很多术语，用眼睛来筛选它们。并选择适当的条款，将花费不满意的时间。

我想收集所有仅一个** 1的倍数的术语。我不关心a的二次或更高的权力，我不关心不包含a的术语。

z和zEx的类型返回以下内容：

print type(z)
print type(zEx)
>>>
<class 'sympy.core.add.Add'>
<class 'sympy.core.mul.Mul'>

有没有人知道如何收集a的倍数，而不是^ 0或^ 2？

tl'dr

其中z（x，y）包含由z和zEx描述的常数a和k以及它们的类型（）：如何从{{{{{}}中删除所有非a项1}}并从表达式中删除z的所有二次或更高项剩下的就是仅包含a的统一力的术语。

Answer 1

最后它只是一个单行。 @asmeurer让我走上正轨（查看本文下面的评论）。这是代码;解释如下：

from sympy import *
from sympy.parsing.sympy_parser import parse_expr
import sys

x, y, k, a = symbols('x y k a')

# modified string: I added a few terms
z = x*(k*a**9) + (k**1)*x**2 - k*a**8 + y*x*(k**2) + y*(x**2)*k**3 + x*(k*a**1) - k*a**3 + y*a**5

zmod = Add(*[argi for argi in z.args if argi.has(a)])

然后zmod是

a**9*k*x - a**8*k + a**5*y - a**3*k + a*k*x

让我们更仔细地看一下：

z.args

只是表达式中所有单个术语的集合（请注意，解析符号也会使事情变得更容易）：

(k*x**2, a**5*y, -a**3*k, -a**8*k, a*k*x, a**9*k*x, k**2*x*y, k**3*x**2*y)

在列表推导中，您可以使用函数a选择包含has的所有术语。然后可以使用Add将所有这些术语粘合在一起，从而为您提供所需的输出。

修改

以上内容返回包含a的所有表达式。如果您只想过滤掉包含a统一功能的表达式，可以使用collect和Mul：

from sympy import * from sympy.parsing.sympy_parser import parse_expr import sys x, y, k, a = symbols('x y k a') z2 = x**2*(k*a**1) + (k**1)*x**2 - k*a**8 + y*x*(k**2) + y*(x**2)*k**3 + x*k*a - k*a**3 + y*a**1 zc = collect(z2, a, evaluate=False) zmod2 = Mul(zc[a], a)

然后zmod2

a*(k*x**2 + k*x + y)

和zmod2.expand()

a*k*x**2 + a*k*x + a*y

这是正确的。

使用您提供的更新z我运行：

z3 = k*(y**2*(a + x) + (a + x)**3/3) - k((2*k*y*(a + x)*(n - 1)*(-k*(y**2*(-a + x) + (-a + x)**3/3) + k*(y**2*(a + x) + (a + x)**3/3)) + y)**2*(-a + k*(n - 1)*(y**2 + (a + x)**2)*(-k*(y**2*(-a + x))))) zc3 = collect(z3.expand(), a, evaluate=False) zmod3 = Mul(zc3[a], a)

然后获取zmod3.expand()：

a*k*x**2 + a*k*y**2

这是你要找的结果吗？

PS：感谢@asmeurer所有这些有用的评论！

Answer 2

除了给出的其他答案外，您还可以使用collect作为字典。

print(collect(zEx,a,evaluate=False)[a])

产生表达式

k*x**2 + k*y**2

Answer 3

要迭代表达式的术语，请使用expr.args。

我不清楚a应该是什么，但collect function可能会做你想要的。

在Sympy中收集表达式的术语

3 个答案: