例如,如果我运行
import sympy
x, y, z = sympy.symbols('x:z')
f = sympy.exp(x + y) - sympy.sqrt(z)
有没有什么f
方法可以用来获取表达式包含的sympy.Symbol
对象的列表或元组?我不希望通过f.args
来解析srepr(f)
或向下解析。
在本例中,g.args[0].args[1].args[0]
给出了Symbol("z")
,而g.args[1].args[0].args
给出了元组(Symbol("x"), Symbol("y"))
,但显然它们是特定于表达式的。
发布于 2015-05-04 04:28:58
您可以使用:
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])
发布于 2020-01-21 22:53:44
一个非常有用的属性是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))
{}
[]
{,}
https://stackoverflow.com/questions/30018977
复制相似问题