如何让SymPy计算含参数的收敛无穷级数？

嘿，我完全懂你遇到的这个尴尬情况——明明知道这个级数收敛，SymPy也应该能算出闭合形式，但直接用带符号参数a计算就是返回原级数，换成具体常数却能出结果，这种“欺骗”的方法确实有点荒谬，不过咱们有正规的解决办法！

问题根源 #

SymPy在处理带符号的无穷级数时，需要明确参数不会导致级数发散的情况。你之前只给a加了real=True，但当a=0时这个级数会变成发散的调和级数平方和，所以SymPy不敢随便应用收敛级数的闭合公式，只能返回未计算的级数。

正确解法：给参数添加足够限制 #

只需要在定义符号a时，额外加上nonzero=True的限制，明确a是不为0的实数，这样SymPy就能放心地调用对应的数学公式了：

from sympy import Symbol, Sum, pi, coth

# 定义不为0的实数参数a
a = Symbol('a', real=True, nonzero=True)
# 定义无穷级数
sum_expr = Sum(1/(n**2 + a**2), (n, 1, oo))
# 求值
result = sum_expr.doit()

print(result)
# 输出：pi*coth(pi*a)/(2*a) - 1/(2*a**2)

你也可以直接用summation函数，效果完全一致：

from sympy import summation

result = summation(1/(n**2 + a**2), (n, 1, oo))
print(result)

验证结果 #

现在你可以把a替换成任意非零实数（比如你之前用的exp(1)），得到的结果和直接代入常数计算的完全匹配：

from sympy import exp

# 替换为exp(1)
print(result.subs(a, exp(1)))
# 和 Sum(1/(n**2 + exp(1)**2), (n,1,oo)).doit() 的结果相同

补充说明 #

这个级数的闭合形式是经典的数学结果，基于余切函数的部分分式展开：

$\sum_{n=1}^\infty \frac{1}{n^2 + a^2} = \frac{\pi}{2a}\coth(\pi a) - \frac{1}{2a^2}$

只要告诉SymPya是非零实数，它就会自动应用这个公式，不用再靠替换常数的“歪招”啦！

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