Python3.11下如何在SymPy的laplace_transform中代入初始条件？

问题背景 #

在Python 3.11版本中，SymPy的laplace_transform函数计算拉普拉斯变换时，结果会保留初始条件相关项，比如y(0)（函数在t=0处的值）、Subs(Derivative(y(t), t), t, 0)（函数一阶导数在t=0处的值）。

示例代码：

from sympy import symbols, Function, laplace_transform

t = symbols('t', real=True)
s = symbols('s', complex=True)
y = Function('y')(t)
laplace_transform(y.diff(t,2)+2*y.diff(t,1)+3*y, t, s)

运行后返回的结果：

s**2*LaplaceTransform(y(t), t, s) + 2*s*LaplaceTransform(y(t), t, s) - s*y(0) + LaplaceTransform(y(t), t, s) - 2*y(0) - Subs(Derivative(y(t), t), t, 0)

需要解决的问题：如何替换结果中的y(0)、Subs(Derivative(y(t),t),t,0)等初始条件项？

解决方案 #

参考相关讨论后，实现了一个自定义函数来批量替换初始条件，代码如下：

from sympy import symbols, Function, laplace_transform, Wild, LaplaceTransform

def apply_initial_conditions(L, ic):
    wf = Wild('f')
    lw = LaplaceTransform(wf, t, s)
    ll = L.find(lw)
    for l in ll: 
        lz = l.match(lw)[wf]
    newL = L
    # 生成待替换的初始条件表达式
    icl = [lz.diff(t, i).subs(t, 0) for i in range(len(ic))]
    # 逐个替换为给定的初始条件值
    for i in range(len(ic)):
        newL = newL.replace(icl[i], ic[i])
    return newL

t = symbols('t', real=True)
s = symbols('s', complex=True)
y = Function('y')(t)
# 计算拉普拉斯变换
Y = laplace_transform(y.diff(t,2)+2*y.diff(t,1)+3*y, t, s)
# 应用初始条件：列表顺序对应y(0)、y'(0)，示例传入[0, 0]
Y = apply_initial_conditions(Y, [0, 0])

函数说明 #

apply_initial_conditions函数的作用是：

• 接收两个参数：L是拉普拉斯变换的结果，ic是初始条件值的列表（顺序为y(0)、y'(0)、y''(0)...）
• 通过Wild匹配出拉普拉斯变换对应的原函数，生成对应的初始条件表达式
• 将这些表达式逐一替换为传入的初始条件值，返回处理后的结果

内容的提问来源于stack exchange，提问作者Andrew