Как я могу использовать lambdify для оценки моей функции? - PullRequest
Новая
       60

Как я могу использовать lambdify для оценки моей функции?

1 голос
/ 21 июня 2019

У меня есть выражение с несколькими переменными, скажем, что-то вроде ниже: 

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 Ответ

4 голосов
/ 21 июня 2019 
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
...