python-3.x 我怎样才能得到一个符号列表在一个sympy表达式?

h9a6wy2h  于 11个月前  发布在  Python
关注(0)|答案(3)|浏览(79)

例如,如果我跑

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

fdbelqdn

fdbelqdn1#

您可以用途:

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

ajsxfq5m

ajsxfq5m2#

一个非常有用的属性是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))


(it值得检查输出)。关于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))

{𝑥} 
[𝑛]
{𝑛,𝑥}

1cosmwyk

1cosmwyk3#

注意JuniorCompressors answer只列出了自由变量
如果你有一个SumProductIntegral或类似的东西,你可能会也可能不想使用.variables属性额外知道积分/求和变量:

In [216]: (x, n) = sympy.symbols("x n")

In [217]: f = sympy.Sum(x, (n, 0, 10))

In [218]: f.free_symbols
Out[218]: {x}

In [219]: f.variables
Out[219]: [n]

字符串

相关问题