与索引变量的Sympy求和

时间:2014-10-16 10:44:04

标签: sum sympy calculus

我尝试使用带有索引变量的Sum创建一个sympy表达式,如前面的解释here 但是,我不能对此表达式进行lambdify并给出一个数组来计算总和。 这不可能吗?

2 个答案:

答案 0 :(得分:2)

也许是这样的?

>>> s=Sum(Indexed('x',i),(i,1,3))
>>> f = lambda x: Subs(s.doit(), [s.function.subs(s.variables[0], j)
... for j in range(s.limits[0][1], s.limits[0][2] + 1)], x).doit()
>>> f((30,10,2))
42

答案 1 :(得分:2)

你可以使用lambdify。只需确保总和的限制与numpy数组的可迭代匹配。

from sympy import Sum, symbols, Indexed, lambdify
import numpy as np

x, i = symbols("x i")
s = Sum(Indexed('x',i),(i,0,3))
f = lambdify(x, s)
b = np.array([1, 2, 3, 4])
f(b)