python - 用 sympy 对方程进行数值计算

Question

我想问一些可以证明是极其简单的事情，但我没有找到如何去做...关键是我想在 python 中使用 sympy 以符号方式定义一些函数，以便使其派生然后用数字表示这个表达式。

这里展示了一个例子:

import numpy as np
from sympy import *

z = Symbol('z')
function = z*exp(z**2)
deriv = diff(function, z)
x = np.arange(1, 3, 0.1) #interval of points
#How can I evaluate numerically this array "x" with the function deriv???

你知道怎么做吗？谢谢!

Answer 1

您可以使用 lambdify使用 numpy 后端:

import numpy as np
from sympy import *

z = Symbol('z')
function = z*exp(z**2)
deriv = diff(function, z)
x = np.arange(1, 3, 0.1) #interval of points
d = lambdify(z, deriv, "numpy")
d(x)
# array([  8.15484549e+00,   1.14689175e+01,   1.63762998e+01,
#          2.37373255e+01,   3.49286892e+01,   5.21825471e+01,
#          7.91672020e+01,   1.21994639e+02,   1.90992239e+02,
#          3.03860954e+02,   4.91383350e+02,   8.07886132e+02,
#          1.35069268e+03,   2.29681687e+03,   3.97320108e+03,
#          6.99317313e+03,   1.25255647e+04,   2.28335915e+04,
#          4.23706166e+04,   8.00431723e+04])