python - 如何获取 sympy 表达式中的符号列表？

Question

例如，如果我运行

import sympy
x, y, z = sympy.symbols('x:z')
f = sympy.exp(x + y) - sympy.sqrt(z)

我可以使用 f 的任何方法来获取表达式包含的 sympy.Symbol 对象的列表或元组吗？我宁愿不必解析 srepr(f) 或通过 f.args 向下解析。

在这种情况下，g.args[0].args[1].args[0] 给了我 Symbol("z")，而 g.args[1].args[0].args 给了我元组 (Symbol("x"), Symbol("y"))，但显然这些是表达式-具体的。

Answer 1

你可以使用:

f.free_symbols

这将返回一组所有自由符号。

例子:

>>> import sympy
>>> x, y, z = sympy.symbols('x:z')
>>> f = sympy.exp(x + y) - sympy.sqrt(z)
>>> f.free_symbols
set([x, z, y])