python - 在 sympy 表达式中查找函数

Question

我正在寻找一种在 Sympy 表达式中查找函数参数的可靠方法。例如，在下面的示例中，如何从 operator 中检索 f？

import sympy

x, y = sympy.symbols('x y')
f = sympy.Function('f')
operator = x*f(x, y).diff(x) + y*f(x, y)

print(operator.free_symbols)
# prints {x, y}

Answer 1

expr.atoms(class) 方法返回 expr 中作为该类实例的所有对象的集合。所以，

operator.atoms(sympy.Function)

将返回 f(x, y)。如果你想忽略函数的参数，使用

[a.func for a in operator.atoms(sympy.Function)]

(对于 SymPy 对象 a，a.func 是它的“函数”部分，a.args 是它的“参数”部分.)

请注意，如果 operator 包含，比如说 sin(x)，那么上面的代码也会包含它。要仅返回 undefined(非内置)函数，请将 sympy.Function 替换为 sympy.function.AppliedUndef 作为 中的类选择器>原子。