python-2.7 - SymPy:评估总和

Question

我需要使用 SymPy 评估表达式:



我希望获得类似的东西



问题是代码

import sympy as sy

n,i,j = sy.symbols('n i j', integer=True)
a = sy.Function('a')

print sy.Sum(sy.Sum(a(j), (j,1,i)), (i, 1, n)).doit()

只打印双倍总和

Sum(a(j), (j, 1, i), (i, 1, n))

Answer 1

您试图获得的答案不正确。内部求和变量为j，用于a(j)，因此无法进一步计算内部和，并且，i为内部和的上限，因此也无法进一步计算外部和.

给出您想要的答案的总和是 Sum(a(i), (j,i,n), (i, 1, n)) ， IE。，

  n     n
 ___   ___
 ╲     ╲
  ╲     ╲   a(i)
  ╱     ╱
 ╱     ╱
 ‾‾‾   ‾‾‾
i = 1 j = i

如果您调用 doit()，SymPy 会计算哪个.顺便说一句，请注意，通过将外部限制作为进一步的参数传递，您可以通过对 Sum 的单次调用来创建双总和。