如何在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])

Answer 2

一个非常有用的属性是atoms

x, y, z = sympy.symbols('x:z')
expr1 = sympy.exp(x + y) - sympy.sqrt(z)
display(expr1.free_symbols)
display(expr1.atoms(sympy.Symbol))

{,,} 
{,,}

除了符号之外，atoms还可以提取其他原子，例如：

display(expr1.atoms(sympy.Function))
display(expr1.atoms(sympy.Number))
display(expr1.atoms(sympy.NumberSymbol))
display(expr1.atoms(sympy.function.AppliedUndef))
display(expr1.atoms(sympy.Mul))
display(expr1.atoms(sympy.Add))

（值得检查输出）。关于gerrit 的回答

n = sympy.Symbol('n')
k2 = sympy.Sum(x, (n, 0, 10))
display(k2.free_symbols)
display(k2.variables)
display(k2.atoms(sympy.Symbol))

{} 
[]
{,}