n元组求和算法

Question

我正在尝试为一些数学公式编写计算机程序，其中我遇到了 n 元组求和，我的意思是 $ \sum \sum \sum \dddot $。我正在寻找有效的算法来用 python 编写代码。例如： $$ \sum_{j_1}^{a_1} \sum_{j_2}^{a_2} \dddot , \sum_{j_i}^{a_i} f(j_1,j_2, \dddot , , j_i) $$ 一个我可以看到的方法是生成一个大小可能为 $ a_1 \dot a_2 \dot \dddot ， a_i $ 的大列表并对其进行迭代。但我认为这不是完美的方法，并且存在更好的解决方案。和的数量不固定，换句话说，表达式中的 sigma 符号的数量可以是任意数量。

Answer 1

如果不深入了解f() 的作用，那么您在优化方面无能为力。你不妨保持你的代码简单-

from itertools import starmap, product

def calcsum(func, intervals):
    ranges = starmap(range, intervals)
    args = product(*ranges)
    return sum(starmap(func, args))