Как я могу использовать sympy sum для элементов списка в python?
Я просто пытаюсь использовать симпатичную сумму по списку элементов, например:
x = [1,2,3,4,5]
expression = sympy.sum(x[i], (i,0,2)) * sympy.sum(x[j], (j,1,3))
и я просто хочу, чтобы он численно оценил это. Я всегда получаю индекс за пределами диапазона, хотя он явно не выходит за пределы диапазона. Есть идеи?
Нет, это даже не с import
Может потому что выдает ошибку? ...
Это дает много ошибок, ни одна из которых не является той, которую вы упомянули. Как спросить и минимальный воспроизводимый пример
Ответы 1
Вам нужно будет построить символьное выражение, представляющее вашу сумму:
from sympy import *
x = IndexedBase("x")
i, j = symbols("i, j")
expr = Sum(x[i], (i,0,2)) * Sum(x[j], (j,1,3))
Затем вы можете заменить x своим «массивом»:
xv = [1,2,3,4,5]
expr.subs(x, Array(xv)).doit()
Обратите внимание, что я преобразовал xv в массив SymPy, иначе процедура не будет работать.
Обновлено: добавлено уточнение
Если ваш список состоит из сотен, тысяч или миллионов записей, то лучше преобразовать символьное выражение в лямбда-функцию, чтобы повысить скорость оценки:
f = lambdify(x, expr)
import inspect
print(inspect.getsource(f))
# def _lambdifygenerated(Dummy_26):
# return (builtins.sum(Dummy_26[i] for i in range(0, 2+1)))*(builtins.sum(Dummy_26[j] for j in range(1, 3+1)))
f(xv)
# 54
С этим можно справиться очень умным способом с помощью целочисленного умножения, как обсуждалось здесь.
@Julien спасибо за ваш ответ, однако он бесполезен. Пример, который я привел выше, ЯВЛЯЕТСЯ минимальным воспроизводимым примером, который вы бы увидели, если бы добавили перед ним "import sympy" и попытались запустить....