python - 具有符号输出的 sympy 方程组

Question

是否可以使用 sympy 求解方程组(线性或非线性)，其中输出为符号？

示例:

 1. f_m = a0 + a1*(-dx) + a2*(-dx)^2 
 2. f_c = a0
 3. f_p = a0 + a1*(dx) + a2*(dx)^2 

求解a2。

通过 Mathematica 命令

解决的办法是

a2 = (1/2)*(f_m - 2*f_c + f_p). 

Answer 1

这是Sympy中的一个基本操作，你应该研究一下文档。只是为了让您开始:

import sympy as sp

f_m, f_c, f_p = sp.var('f_m, f_c, f_p')
a0, a1, a2 = sp.var('a0:3')
dx = sp.var('dx')

eq1 = sp.Eq(f_m, a0 + a1*(-dx) + a2*(-dx)**2)
eq2 = sp.Eq(f_c, a0)
eq3 = sp.Eq(f_p, a0 + a1*(dx) + a2*(dx)**2 )


sp.linsolve([eq1, eq2, eq3], (a0, a1, a2))
# sp.solve([eq1, eq2, eq3], (a0, a1, a2))  # also works

{(f_c, (-f_m + f_p)/(2*dx), (-2*f_c + f_m + f_p)/(2*dx**2))}