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