对多个符号变量使用 Lambdify

Question

下面的 python 脚本计算表达式中的第一项， 4.939e-3xAxB+8.7989 at (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))