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

2021-03-01 01:58:51 发布

您现在位置:Python中文网/ 问答频道 /正文

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

from sympy import *

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

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

print(a)

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

1条回答
网友
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]

相关问题