如何使用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