如何在sympy表达式中获取符号列表?

时间:2015-05-03 20:23:45

标签: python python-3.x sympy

例如,如果我运行

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")),但显然这些是特定于表达式的。

3 个答案:

答案 0 :(得分:34)

您可以使用:

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])

答案 1 :(得分: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))

(值得检查输出)。 关于answer by gerrit

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

{?} 
[?]
{?,?}

答案 2 :(得分:1)

请注意,JuniorCompressors answer仅列出自由变量

如果您有SumProductIntegral或类似内容,您可能希望或不想使用{{1}另外知道积分/求和变量属性:

.variables