如何使用lambdify评估我的功能?

时间:2019-06-20 21:22:41

标签: python numpy symbols sympy lambdify

我有一个带有多个变量的表达式,让我们说如下:

import numpy as np
import sympy as sym
from sympy import Symbol, Add, re, lambdify
x = sym.Symbol('x')
y = sym.Symbol('y')
z = sym.Symbol('z')

F = x+ y +z

我有三个变量列表,如下所示:

x = [3, 2 ,3]   
y = [4, 5 , 6] 
z = [7, 10 ,3]

我想为变量的每个元素评估函数。 我知道我可以定义如下内容:

f_dis = lambdify([x, y, z], x + y + z, 'numpy')
d = f_dis(3, 4, 7)
print ( "f_dis =", d)

给我14个期望的结果。但是如何将x,y和z作为三个列表传递(而不是分别编写元素),并得到如下结果:

[14, 17, 12]

根据以下说明,使用lambdify似乎是评估函数的更有效方法: https://www.sympy.org/scipy-2017-codegen-tutorial/notebooks/22-lambdify.html

谢谢。

1 个答案:

答案 0 :(得分:4)

import sympy as sp

x = sp.Symbol('x')
y = sp.Symbol('y')
z = sp.Symbol('z')

X = [3, 2 ,3]   
Y = [4, 5 , 6] 
Z = [7, 10 ,3]

values = list(zip(X, Y, Z))


f_dis = sp.lambdify([x, y, z], x + y + z, 'numpy')
ans = [f_dis(*value) for value in values]
for d in ans:
    print ( "f_dis =", d)

这将为您提供:

f_dis = 14
f_dis = 17
f_dis = 12