如何在SymPy符号间定义约束？解决卷积分段函数问题

方法一：用refine结合符号假设过滤分段分支 #

SymPy的Q模块可声明符号间的逻辑约束，配合refine工具能自动剔除分段函数中不符合约束的分支，直接得到目标结果。

具体操作：

• 导入Q和refine工具
• 定义所有tau互不相等的联合假设
• 每次卷积运算后，用refine将假设应用到结果上

修改后的完整代码：

import sympy as sp
from sympy import Q, refine
from sympy.plotting import plot

def convolve(f, g, t, lim_low, lim_up):
    tau = sp.symbols('tau', real=True, positive=True)
    h = sp.integrate(f.subs(t, tau) * g.subs(t, t - tau),
                  (tau, lim_low, lim_up))
    return h

# 定义符号
tau_1, tau_2, tau_3 = sp.symbols('tau_1, tau_2, tau_3', real=True, positive=True)
t = sp.symbols('t', real=True)
# 声明所有tau互不相等的约束假设
assumptions = Q(tau_1 != tau_2) & Q(tau_1 != tau_3) & Q(tau_2 != tau_3)

f1 = sp.exp( -t/tau_1 )
f2 = sp.exp( -t/tau_2 )
f3 = sp.exp( -t/tau_3 )

# 第一次卷积并应用约束
h12 = convolve(f1, f2, t, lim_low=0, lim_up=t).simplify()
h12 = refine(h12, assumptions)
print('First convolution (after applying assumptions):')
display(h12)

# 第二次卷积，直接使用已简化的h12
h123 = convolve(h12, f3, t, lim_low=0, lim_up=t).simplify()
h123 = refine(h123, assumptions)
display(h123)

方法二：修改积分参数跳过分段生成 #

若你明确所有tau互不相等，可在sp.integrate中添加conds='none'参数，让SymPy直接返回非分段表达式，省去后续处理分段的步骤。

修改后的卷积函数：

def convolve(f, g, t, lim_low, lim_up):
    tau = sp.symbols('tau', real=True, positive=True)
    # 添加conds='none'参数，跳过分段函数生成
    h = sp.integrate(f.subs(t, tau) * g.subs(t, t - tau),
                  (tau, lim_low, lim_up), conds='none')
    return h

使用该函数后，直接运行原代码的卷积步骤，得到的结果即为tau不等时的表达式，无需额外处理分段部分。

注意：此方法仅适用于确定所有tau互不相等的场景，否则当tau相等时，结果会出现分母为0的无意义表达式。

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