Проверьте, есть ли выражение sympy в функции

Question

Есть ли способ программно определить, содержит ли выражение SymPy sympy.Function? Например, из a в

from sympy import * 

f = Function('f') 
g = Function('g') 

a = f(3) + g(5) + 1.0 

print(a) 

Я хотел бы получить объекты f и g.

Answer 1

Используйте atoms, чтобы найти все атомы данного типа:

import sympy as sy 

f = sy.Function('f') 
g = sy.Function('g') 

a = f(3) + g(5) + 1.0 

print(a) 
# f(3) + g(5) + 1.0 
print(a.atoms(sy.Function)) 

дает

set([f(3), g(5)]) 

Чтобы получить сам Function,

map(type, a.atoms(sy.Function)) 

дает

[f, g]