Question

我想使用sympy库中嵌入的椭圆函数，但是当我尝试lambdify椭圆（f（z））时出现问题，其中f是一个符号函数：

import sympy as sym
import numpy as np

r = sym.S('r')
_f = 500.0*r
f=sym.lambdify(r,_f,'numpy')

f(np.arange(0,1.1,0.1) )

输出正确：

array([   0.,   50.,  100.,  150.,  200.,  250.,  300.,  350.,  400.,
    450.,  500.])

然后我尝试f2（f（r））：

_f2 = _f/10
f2= sym.lambdify(r,_f2,'numpy')

f2(np.arange(0,1.1,0.1) )

它有效：

array([  0.,   5.,  10.,  15.,  20.,  25.,  30.,  35.,  40.,  45.,  50.])

但是当我尝试时：

_ek=sym.elliptic_k(_f)
ek=sym.lambdify(r,_ek,'numpy')

ek(0)

我明白了：

Traceback (most recent call last):

File "<ipython-input-17-35ab6a3fc36f>", line 1, in <module>
EK(0)

File "<string>", line 1, in <lambda>

NameError: name 'elliptic_k' is not defined

任何想法如何解决？当然我想尽可能不重写所有的elliptic_k函数。谢谢

Answer 1

答案在lambdify的文档中：

Attention: Functions that are not in the math module will throw a name error when the lambda function is evaluated!。

因此，您需要将lambdify相应模块的名称（sympy而不是numpy传递给elliptic_k。

ek=sym.lambdify(r,_ek,'sympy')

Answer 2

NumPy没有椭圆函数，但是SciPy does。 lambdify尚未自动包含SciPy功能，因此您需要执行

lambdify(r, _ek, modules=['numpy', {'elliptic_k': scipy.special.ellipk}])

有关详细信息，请参阅the lambdify documentation。

Answer 3

sympy.functions.elliptic_k也许？

如何在同情中对椭圆函数进行lambdify

3 个答案: