Привыкание индексов индексированных терминов в SymPy сумму

Question

Скажем, у меня есть переменные x и y, которые индексируются

from sympy.tensor import IndexedBase 
x = IndexedBase('x')` 

И у меня есть выражение, как e = x[1] y[2] + x[5] y[10] . Я хочу найти все индексы, используемые каждым из x и y. Я ищу функцию, которая может работать следующим образом: e.indices(y) = [2, 10] и e.indicies(x) = [1, 5]

Есть ли способ, которым я могу перебирать термины x[i] y[j]? И если да, есть ли способ разделить продукт на термины и для каждого из них вытащить, какая буква используется и какой индекс появляется?

Answer 1

Следующая должны получить вы движетесь в правильном направлении:

>>> from sympy.tensor import IndexedBase, Indexed 
>>> from sympy import sift 
>>> x = IndexedBase('x') 
>>> y = IndexedBase('y') 
>>> e = x[1]* y[2] + x[5]* y[10] 
>>> e.atoms(IndexedBase) 
set([y, x]) 
>>> e.atoms(Indexed) 
set([x[5], y[10], x[1], y[2]]) 
>>> sifted = sift(_,lambda i: i.base) 
>>> sifted[x] 
[x[5], x[1]] 
>>> sifted[y] 
[y[10], y[2]] 
>>> [i.indices for i in _] 
[(10,), (2,)] 
>>> flatten(_) 
[10, 2]