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