如何在SymPy中展开和的乘积并引入哑指标？

开发者阿泽

2026-6-26

解决SymPy展开求和乘积的问题

要将$(\sum_{i=1}^N \sum_{j=1}^N A_i A_j)^2$展开为四重求和形式，你需要使用SymPy专门处理多重求和乘积的工具，具体方案如下：

方法1：使用`expand_multisum`函数

显式将平方转换为两个求和的乘积，再调用expand_multisum展开：

from sympy import symbols, Dummy, IndexedBase, Sum, expand_multisum

i, j, k, l = symbols('i, j, k, l', cls=Dummy)
A = IndexedBase('A')
N = symbols('N', integer=True, positive=True)
s = Sum(A[i]*A[j], (i, 1, N), (j, 1, N))
expr = s * s  # 写成两个求和的乘积而非平方形式
expanded_expr = expand_multisum(expr)

执行后expanded_expr即为$\sum_{i=1}^N \sum_{j=1}^N \sum_{k=1}^N \sum_{l=1}^N A_i A_j A_k A_l$。

方法2：使用`expand`的`multisum`参数

直接对平方形式的表达式调用expand并指定multisum=True参数：

from sympy import symbols, Dummy, IndexedBase, Sum

i, j, k, l = symbols('i, j, k, l', cls=Dummy)
A = IndexedBase('A')
N = symbols('N', integer=True, positive=True)
s = Sum(A[i]*A[j], (i, 1, N), (j, 1, N))
expr = s**2
expanded_expr = expr.expand(multisum=True)

此方法同样能得到目标的四重求和表达式。