python - 对多个符号变量使用 Lambdify

Question

下面的 python 脚本计算表达式中的第一项，
4.939e-3xAxB+8.7989 在 (A,B) = (1.0,1.0):

 import sympy
 from sympy import *

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

输出为:0.004939

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

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

但这会返回错误: NameError: global name 'B' is not defined
有人可以解释一下吗？

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