如何用Sympy对符号集合j按集合i的范围求和？

现有两个各包含m个元素的集合i、j，需要对由i和j元素构成的表达式（比如元素乘积或KroneckerDelta函数）进行求和，要求j的每个元素都遍历集合i的所有元素[i₁,i₂,...,iₘ]。但SymPy的Sum函数要求求和变量以(symbol, start, stop)形式定义，默认采用整数步长，无法直接实现这种自定义集合的遍历求和。

示例场景代码：

from sympy import *
i = symbols('i1,i2,i3,i4')  # m=4的情况
j = symbols('j1,j2,j3,j4')

通过排列组合构建的乘积表达式：

from itertools import permutations as perm
c = list(perm(range(4),2))
a,b = c[0]
expr = i[a]*j[b]
for a,b in c[1:]:
    expr += i[a]*j[b]
print(expr)
# 输出：i1*j2 + i1*j3 + i1*j4 + i2*j1 + i2*j3 + i2*j4 + i3*j1 + i3*j2 + i3*j4 + i4*j1 + i4*j2 + i4*j3

实际需求的KroneckerDelta求和场景：

from sympy import KroneckerDelta as KD
expr = KD(i[0],j[1])
print(expr)
# 输出：KroneckerDelta(i1,j2)
# 期望求和后得到：1 + KroneckerDelta(i1,i2) + KroneckerDelta(i1,i3) + KroneckerDelta(i1,i4)

方法1：手动遍历替换求和（适合小m场景） #

直接遍历集合i的元素，将j的每个变量依次替换为i中的元素后累加求和，这种方式直观且容易实现，适合元素数量较少的场景。

代码示例（KroneckerDelta场景）：

from sympy import KroneckerDelta as KD, symbols

i = symbols('i1,i2,i3,i4')
j1 = symbols('j1')

expr = KD(i[0], j1)
# 遍历i的所有元素替换j1并求和
sum_result = sum(expr.subs(j1, elem) for elem in i)
print(sum_result)
# 输出：1 + KroneckerDelta(i1, i2) + KroneckerDelta(i1, i3) + KroneckerDelta(i1, i4)

方法2：结合Piecewise自定义求和范围（通用场景） #

通过Piecewise构造求和变量的取值条件，限制求和变量只能取集合i中的元素，再配合Sum函数从负无穷到正无穷求和（仅保留符合条件的项），这种方法更贴近SymPy原生求和的写法。

代码示例：

from sympy import Sum, KroneckerDelta as KD, symbols, Piecewise, Eq

i = symbols('i1,i2,i3,i4')
j1 = symbols('j1')
expr = KD(i[0], j1)

# 构造j1只能取i中元素的条件
condition = Eq(j1, i[0]) | Eq(j1, i[1]) | Eq(j1, i[2]) | Eq(j1, i[3])
# 仅对满足条件的j1项求和
sum_result = Sum(expr * Piecewise((1, condition), (0, True)), (j1, -oo, oo)).doit()
print(sum_result)
# 输出：1 + KroneckerDelta(i1, i2) + KroneckerDelta(i1, i3) + KroneckerDelta(i1, i4)

方法3：封装自定义求和函数（批量处理场景） #

如果需要对多个j变量进行批量求和，可以封装一个通用函数，自动处理每个变量的遍历替换逻辑，提升代码复用性。

代码示例（多变量求和场景）：

from sympy import symbols, KroneckerDelta as KD
from itertools import permutations

def sum_over_set(expr, var, target_set):
    """
    对变量var求和，遍历target_set中的所有元素
    """
    return sum(expr.subs(var, elem) for elem in target_set)

# 构造多变量表达式
i = symbols('i1,i2,i3,i4')
j = symbols('j1,j2,j3,j4')
c = list(permutations(range(4), 2))
expr = sum(i[a]*j[b] for a,b in c)

# 对每个j变量依次执行集合遍历求和
for var in j:
    expr = sum_over_set(expr, var, i)

print(expr)
# 输出所有j变量替换为i元素后的求和结果

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