在 Python 中声明多个符号变量

Question

当我声明一个符号变量时，它可以工作：

>>> from sympy import var
>>> x = var('x')
>>> x + 2
x + 2

现在，出于我的目的，我需要多个变量，例如 s0, s1, ..., s9，我还需要像 s0 + 1、s2 - s1 等这样的操作。

代码是什么？这对我不起作用（编辑：我的意思是我可以这样做，是的，但为此我需要大量更改现有代码）：

>>> from sympy import symbols
>>> s = symbols('s0:9'); s
(s0, s1, s2, s3, s4, s5, s6, s7, s8)
>>> s[0] + 1
s0 + 1

EDIT2：s0, s1, s2, s3, s4, s5, s6, s7, s8, s9 = symbols('s0:10') 可以，但变量的数量不固定。

Answer 1

首先，您应该使用symbols 而不是var。 var 可以将符号注入命名空间，并且只能以交互方式使用。

如果你想要任意数量的符号，你需要numbered_symbols 函数，它产生一个迭代器。 Here 是文档。一个例子

>>> N = numbered_symbols('s')
>>> for s, _ in zip(N, range(10)):
...     print(s)
s0
s1
s2
s3
s4
s5
s6
s7
s8
s9

Answer 2

这很好用，不知道你遇到了什么问题：

>>> from sympy import symbols
>>> s = symbols('s0:10')
>>> s
(s0, s1, s2, s3, s4, s5, s6, s7, s8, s9)
>>> s[0] + 1
s0 + 1

如果你想将每个设置为一个变量，你可以使用多重赋值：

s0, s1, s2, s3, s4, s5, s6, s7, s8, s9 = symbols('s0:10')

这相当于：

s0, s1 = Symbol('s0'), Symbol('s1') # and s2:s9 as well