SymPy中带符号边界的符号求和未展开问题及优化方案咨询

问题背景 #

使用SymPy求解带求和符号的方程组时，Sum_x未自动展开为具体数值，仍保留求和表达式形式。原代码及输出如下：

原代码 #

from sympy import *
from sympy.abc import i

# Symbols
x = IndexedBase('x')
sum_x = Symbol("Sum_x")
sym_bound = Symbol("upper_bound")

concrete_bound = 5
x_constraints = [Eq(x[idx], idx**2) for idx in range(concrete_bound)]

sym_bound_eq = Eq(sym_bound, 5)
equation = Eq(sum_x, Sum(x[i], (i, 0, sym_bound - 1)).doit())

solution = solve(x_constraints + [equation, sym_bound_eq])
print(solution)

实际输出 #

{Sum_x: Sum(x[i], (i, 0, upper_bound - 1)), upper_bound: 5, x[0]: 0, x[1]: 1, x[2]: 4, x[3]: 9, x[4]: 16}

原因分析 #

solve函数优先保留符号表达式的结构，不会自动将求解得到的变量值代入求和式并展开。原代码中调用.doit()时，sym_bound仍是未赋值的符号变量，因此求和无法展开。

优化解决方案 #

无需手动逐个代入变量，可利用求解得到的字典一次性完成所有变量的替换，再展开求和：

优化代码 #

from sympy import *
from sympy.abc import i

# 定义符号
x = IndexedBase('x')
sum_x = Symbol("Sum_x")
sym_bound = Symbol("upper_bound")

concrete_bound = 5
x_constraints = [Eq(x[idx], idx**2) for idx in range(concrete_bound)]

sym_bound_eq = Eq(sym_bound, 5)
# 此处先不调用doit()，留到求解后再处理
equation = Eq(sum_x, Sum(x[i], (i, 0, sym_bound - 1)))

# 以字典形式获取解（dict=True返回字典列表）
solution = solve(x_constraints + [equation, sym_bound_eq], dict=True)[0]

# 一次性代入所有已知变量并展开求和
sum_x_result = solution[sum_x].subs(solution).doit()
print(sum_x_result)  # 输出：30

代码说明 #

1. 使用dict=True让solve返回字典格式的解，方便直接代入表达式；
2. 调用.subs(solution)将解字典中所有变量（包括sym_bound和x的各索引值）一次性替换到求和表达式中；
3. 最后用.doit()触发求和运算，得到具体数值结果。

这种方式避免了手动遍历变量代入的繁琐操作，代码更简洁且扩展性更强。

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