gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-28 16:26:47 25 4
gpt4 key购买 nike

有没有办法以编程方式确定 SymPy 表达式是否包含 sympy.Function?例如,从 a

from sympy import *

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

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

print(a)

我想获取对象 fg

最佳答案

使用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]

关于python - 检查 sympy 表达式是否包含函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35984805/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com