在python中操作函数

时间:2016-11-09 15:56:27

标签: python function equation algebra

我希望能够在另一个函数中使用一个函数,在本例中是函数nu用于被积函数的表达式。目前这是我的代码:

from trapezium import trap
import scipy as sp

# mass_enc returns the mass enclosed within a radius R
# nu is the density function

def mass_enc(q, R):
    integrand = lambda r: 4 * sp.pi * r**2 * q(r)
    return trap(integrand, 0, R, 100)

def nu(a):
    return a

print(mass_enc(nu, 10))

然而这似乎很混乱 - 有更好的方法吗?

0 个答案:

没有答案