如何在Python循环中实现SymPy符号变量替换求和

我正在进行包含多个符号变量x、t、n的积分计算，得到关于x、t、n的函数g后，希望通过for循环替换变量n的取值实现求和操作，但g.subs({n:c})命令无法正常工作。需要将n依次替换为1到10的数值（数学上即求n从1到10时g(x,t)的和），请问当函数包含三个未知变量时，SymPy是否有可替换符号变量n的级数函数？

以下是最小可复现代码（MWE）：

from sympy import pi, series
#from spb import *
#from mpmath import nsum, inf

import sympy as sm
x = sm.symbols("x")
t = sm.symbols("t")
n = sm.symbols("n", integer=True)

L = 20
f1 = (2/L)*x*sm.sin(n*pi*x/20)
f2 = (2/L)*(20-x)*sm.sin(n*pi*x/20)
sm.pretty_print(sm.integrate(f1,(x,0,10)))
sm.pretty_print(sm.integrate(f2,(x,10,20)))

D = 0.475
g = (f1+f2)*sm.sin(n*pi*x/20)*sm.exp(-(n**2)*(pi**2)*D*t/400)

sm.pretty_print(g)

gn = g.subs({n:1})
sm.pretty_print(gn)

for c in range(10):
    g += g.subs({n:c})
    print(g)

问题根源 #

原代码的循环逻辑有误：每次循环都直接修改了g变量本身，导致后续累加的是已经被修改后的g（包含之前的求和项），而不是基于原始g替换n得到的新项，这会导致结果完全错误。

方法1：使用SymPy内置的summation函数直接求和 #

SymPy提供了summation函数，可以直接对符号变量在指定区间内求和，这是最简洁的方式：

from sympy import pi, summation
import sympy as sm

x = sm.symbols("x")
t = sm.symbols("t")
n = sm.symbols("n", integer=True)

L = 20
f1 = (2/L)*x*sm.sin(n*pi*x/20)
f2 = (2/L)*(20-x)*sm.sin(n*pi*x/20)
D = 0.475

# 定义原始的g函数
g = (f1+f2)*sm.sin(n*pi*x/20)*sm.exp(-(n**2)*(pi**2)*D*t/400)

# 计算n从1到10的求和
total_sum = summation(g, (n, 1, 10))
sm.pretty_print(total_sum)

方法2：修正for循环的写法 #

如果坚持用for循环，需要先保存原始的g函数，每次循环基于原始g替换n的值，再累加到结果变量中：

from sympy import pi
import sympy as sm

x = sm.symbols("x")
t = sm.symbols("t")
n = sm.symbols("n", integer=True)

L = 20
f1 = (2/L)*x*sm.sin(n*pi*x/20)
f2 = (2/L)*(20-x)*sm.sin(n*pi*x/20)
D = 0.475

# 保存原始的g，避免后续修改影响替换
original_g = (f1+f2)*sm.sin(n*pi*x/20)*sm.exp(-(n**2)*(pi**2)*D*t/400)

# 初始化求和结果
total_sum = sm.sympify(0)

# 循环n从1到10
for c in range(1, 11):
    term = original_g.subs({n: c})
    total_sum += term

sm.pretty_print(total_sum)

补充说明 #

• summation函数支持符号求和，只要被求和的表达式可以被SymPy处理，就能直接得到简洁的符号结果，推荐优先使用。
• 替换变量时，必须确保基于未被修改的原始表达式进行替换，否则会引入错误的累加逻辑。

内容的提问来源于stack exchange，提问作者Freya the Goddess