获取 sympy 解决方案的某些部分

Question

我在表格上有一个巨大的象征性同情表达

expression = factor * (f1*a + f2*b + f3*c + f4*d + f5*e)

其中 a 到 e 的所有因素均由多项组成。即：

a = exp(2x) + exp(3x) + sin(Ix).

我想在表单上创建一个数组

array = factor * [a,b,c,d,e]

但是没有看到一个切肉刀的方法来做到这一点。我尝试使用 factor 函数，但它只给出了上面 "expression" 形式的表达式。

到现在为止我用过[=14=]

print(expression)

然后对因子 a 到 e 进行了一些强力复制粘贴。由于我将获得比本例中更多术语的表达式，因此我想在没有复制粘贴例程的情况下执行此操作。有什么想法吗？

Answer 1

这是一个简单的示例，您可以推断出更多项

import sympy as sp

x = sp.var('x')
f1, f2 = sp.symbols('f1:3')
factor = sp.symbols('factor')

a = x**2 + sp.sin(x) + sp.exp(sp.I * x)
b = sp.log(x)/(x+1)**2

# example expression:
expression = (factor * (f1 * a + f2 * b)).expand()
print(expression)

# collect coefficients of f1 and f2
coeffs = sp.collect(expression.expand(),[f1,f2], evaluate=False)
print(coeffs)

# show the coefficients w/o the factor factor
[(coeffs[f]/factor).simplify() for f in (f1,f2)]

f1*factor*x**2 + f1*factor*exp(I*x) + f1*factor*sin(x) + f2*factor*log(x)/(x**2 + 2*x + 1)
{f2: factor*log(x)/(x**2 + 2*x + 1), f1: factor*x**2 + factor*exp(I*x) + factor*sin(x)}

[x**2 + exp(I*x) + sin(x), log(x)/(x**2 + 2*x + 1)]

获取 sympy 解决方案的某些部分

Fetch certain parts of sympy solution

python

arrays

math

sympy

factors