.coeff() 没有给出正确答案 Python Sympy

Question

我试图在 sympy 中找到表达式的系数。

我编写了一个代码，但它给出了错误的输出。第一个（下面给出的）代码是我尝试编写程序的方式（尽管它并不完全正确）。

import sympy as sp

a0, a1, a2, a3, x = sp.symbols("a_0 a_1 a_2 a_3 x")
v = a0 + a1*x + a2*x**2 + a3*x**3
display(v)

print("========================================================================")

v1, θ1, v2, θ2, L = sp.symbols("v1 θ_1 v2 θ_2 L")


v = v.subs([(a0, v1), (a1, θ1*v2), (a2, L*v2 + v1 + θ1/θ2), (a3, θ1*θ2)])
display(v)

print("========================================================================")

display(v.coeff(x, 2))
display(v.coeff(v2))
display(v.coeff(θ2))

这是输出。

在这里，即使我用其他符号组合替换了 a0, a1, a2 and a3，我仍然可以调用 .coeff() 函数并得到一个系数（尽管在第二个 v2)

我有另一个代码，我可以在其中做更多的事情

import sympy as sp

a0, a1, a2, a3, x, L = sp.symbols("a_0 a_1 a_2 a_3 x L")
v1, θ1, v2, θ2 = sp.symbols("v1 θ_1 v2 θ_2")

v = a0 + a1*x + a2*x**2 + a3*x**3 # Equation
θ = sp.diff(v, x)

display(v, θ)

print("=============================================================")

v_1 = v.subs(x, 0) # substituting values for x 
θ_1 = θ.subs(x, 0)
v_2 = v.subs(x, L)
θ_2 = θ.subs(x, L)

display(v_1, θ_1, v_2, θ_2)
# now we got 4 equations in terms of a0, a1, a2 and a3

print("==============================================================")

A = sp.solve((v_1 - v1, θ_1 - θ1, v_2 - v2, θ_2 - θ2), a0, a1, a2, a3) 
# solving system of 4 equations in terms of a0, a1, a2 and a3

display(A)

print("==============================================================")

A0, A1, A2, A3 = A[a0], A[a1], A[a2], A[a3]

display(A0, A1, A2, A3)

print("==============================================================")

v = v.subs([(a0, A0), (a1, A1), (a2, A2), (a3, A3)])
display(v)

print("==============================================================")

display(v.coeff(x, 2)) # this works for x, x**2, x**3 etc

print("==============================================================")

display(v.coeff(v1), v.coeff(θ1), v.coeff(v2), v.coeff(θ2)) # THIS GIVES WRONG ANSWER

这是新代码的输出。除了最后一行输出错误之外，此代码中的所有内容都是正确的。

在这里，当我为 v1, θ1, v2, θ2 调用 .coeff() 时，我没有得到正确的答案。

以上为正确答案：

我正试图得到这样的答案。有人可以帮忙吗？我认为问题出在 .solve() 及其输出中。有人可以帮助我吗？

谢谢。

Answer 1

我的错。应该在 .coeff().

之前完成 v = v.expand()

.coeff() 没有给出正确答案 Python Sympy

.coeff() does not give correct answer Python Sympy

python

linear-equation

solver

sympy

coefficients