例如,如果我运行
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"))
,但显然这些是特定于表达式的。
答案 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仅列出自由变量。
如果您有Sum
,Product
,Integral
或类似内容,您可能希望或不想使用{{1}另外知道积分/求和变量属性:
.variables