Sympy将变量代入到变量

发现sympy很有意思，不仅可以把常数代入变量计算数值结果，还可以将变量代入变量。

• 而且发现，要代入的变量可以是局部变量也可以是全局变量。

下面举个具体的例子。

例子

代入常数

from sympy import *

def test_subs(a):
    x = symbols('x')
    f = x + 1
    f = f.subs(x, a)
    return f

print(test_subs(1))

输出:

2

代入全局变量

from sympy import *

z = symbols('z')
a = z ** 2

def test_subs(a):
    x = symbols('x')
    f = x + 1
    f = f.subs(x, a)
    return f

print(test_subs(a))

输出:

z**2 + 1

代入局部变量

代入局部变量也是可以的，目前来看，对结果没有影响。

from sympy import *

def get_a():
    z = symbols('z')
    a = z ** 2
    return a

def test_subs(a):
    x = symbols('x')
    f = x + 1
    f = f.subs(x, a)
    return f

print(test_subs(get_a()))

输出：

z**2 + 1

可以看到结果与代入全局变量是一样的。

分享 补充完善

12-07-2021

wiki