python - 如何使用 lambdify 评估我的函数？

Question

我有一个包含多个变量的表达式，让我们说如下：

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

谢谢。

Answer 1

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