Sympy：從表達式中獲取函數

Question

要從sympy表達式中獲取所有變量，可以在表達式上調用.free_symbols 。 我想檢索表達式中使用的所有函數 。 例如，從y in

from sympy import *

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

x = Symbol('x')

y = f(x) + 2*g(x)

我想得到f和g 。

任何提示？

Answer 1

atoms可以解決問題：

for f in y.atoms(Function):
    print(f.func)

Answer 2

對於所有函數，使用atoms(Function) 。

In [40]: (f(x) + sin(x)).atoms(Function)
Out[40]: set([f(x), sin(x)])

對於僅未定義的函數，請使用atoms(AppliedUndef) 。

In [41]: from sympy.core.function import AppliedUndef

In [42]: (f(x) + sin(x)).atoms(AppliedUndef)
Out[42]: set([f(x)])