我希望能够在另一个函数中使用一个函数,在本例中是函数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))
然而这似乎很混乱 - 有更好的方法吗?