将Lambdify用于多个符号变量

时间:2017-10-25 17:36:55

标签: python sympy

下面的python脚本评估表达式中的第一个术语, 4.939e-3xAxB + 8.7989 (A,B)=(1.0,1.0)

 import sympy
 from sympy import *

 A = sympy.Symbol(chr(65))
 B = sympy.Symbol(chr(66))
 F = 4.939e-3*A*B+8.7989
 G = str(F).split("+")[0]
 H = lambdify([chr(65),chr(66)], G, "numpy")
 print H(1,1)

输出为:0.004939

现在,下面的代码旨在实现相同的目标:

A = sympy.Symbol(chr(65))
B = sympy.Symbol(chr(66))
F = 4.939e-3*A*B+8.7989
G = str(F).split("+")[0]
H = lambdify(chr(65), G, "numpy")
I = lambdify(chr(66), H(1), "numpy")
print I(1)

但是这会返回错误: NameError: global name 'B' is not defined

有人可以解释一下吗?

1 个答案:

答案 0 :(得分:1)

您在多个实例中混淆字符串和符号。任何这一点的唯一原因是SymPy将sympify应用于许多输入(sympify("A")==Symbol("A"))和Python的鸭子输入。具体来说,G是一个字符串,当它应该是一个符号表达式时,传递给lambdify的所有第一个参数都是字符串或字符串列表,当它们应该是符号或列表时。

您的第一个代码的干净版本将是:

from sympy.abc import A,B
import sympy

G = 4.939e-3*A*B # now this is a symbolic expression, not a string
H = lambdify( [A,B], G, "numpy" )
print(H(1,1))

第二段代码也一样:

from sympy.abc import A,B
import sympy

G = 4.939e-3*A*B # now this is a symbolic expression, not a string
H = lambdify( A, G, "numpy" )
I = lambdify( B, H(1), "numpy" )
print(I(1))