检查sympy表达式是否包含函数

Question

有没有一种方法可以通过编程来确定一个SymPy表达式是否包含sympy.Functions？例如，从中的a

from sympy import *

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

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

print(a)

我想得到对象f和{}。在

Answer 1

使用^{}查找给定类型的所有原子：

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))

收益率

^{pr2}$

为了得到Function本身

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

收益率

[f, g]