递归地与Python中的Simpson规则集成

时间:2015-04-25 06:13:42

标签: python numpy numerical-methods

我想用Python来实现Simpson Integration。如果我不需要它自动收敛(这里我需要abs(result - my_expect) < 0.001),这并不难。 但我想要一个用Python编写的自动融合程序。所以我尝试使用我从 SICP 中学到的方法 - 以递归方式进行。

# imports
from __future__ import division
import numpy as np

# constants
MU = 50
SIGMA = 15
X0 = 0
XN = 100


# interface functions
def f(x):
    """Integrand.

    It will be used for Lagrangian interpolation and
    calculating corresponding function value in function
    `improve_precision`.

    """
    y = (1 / (SIGMA * np.sqrt(2 * np.pi))
         * np.exp(-1/2 * (x-MU)**2 / SIGMA**2))
    return y


# classes
class Integration:
    def __init__(self, start, end):
        self.start = start
        self.end = end
        self.a = (self.end - self.start) / 2
        self.h = 2 * self.a
        self.rp = 0
        self.s = 0
        self.i = 1

    def integrate(self):
        return self.integrate_iter(1, 0, 0.001)

    def integrate_iter(self, expectation, summation, accuracy):
        if accuracy > abs(summation - expectation):
            return summation
        else:
            self.integrate_iter(expectation,
                                self.improve_precision(self.i+1), 
                                accuracy)

    def improve_precision(self, ii):
        rc = np.sum([f(-self.a - self.h/2 + k*self.h)
                     for k in range(1, 2**ii + 1)])
        self.s = self.h/6 * (self.rp + 4*rc)
        self.h /= 2
        self.rp = self.rp + 2*rc
        return self.s

calc = Integration(X0, XN)
print calc.integrate()

当我运行它时,出现了很多错误:

  File "/Users/*/integrate.py", line 72, in <module>
    print calc.integrate()
  File "/Users/*/integrate.py", line 56, in integrate
    return self.integrate_iter(1, 0, 0.001)
  File "/Users/*/integrate.py", line 62, in integrate_iter
    self.integrate_iter(expectation, self.improve_precision(self.i+1), accuracy)
  File "/Users/*/integrate.py", line 62, in integrate_iter
    self.integrate_iter(expectation, self.improve_precision(self.i+1), accuracy)
  ...
  File "/Users/*/integrate.py", line 65, in improve_precision
    rc = np.sum([f(-self.a - self.h/2 + k*self.h) for k in range(1, 2**ii + 1)])
  File "/usr/local/lib/python2.7/site-packages/numpy/core/fromnumeric.py", line 1708, in sum
    if isinstance(a, _gentype):
RuntimeError: maximum recursion depth exceeded while calling a Python object
Process finished with exit code 1

那我该怎么办?有没有办法让它正确?

编辑:

好吧,我对函数integrate_iter进行了更改,并添加了一个函数is_rough。我的最终代码是这样的:

...
    def integrate(self):
        """A function encapsulate calculating process."""
        integrate_value = self.integrate_iter(0)
        interval_count = self.i
        return integrate_value, interval_count

    def integrate_iter(self, summation):
        """Main part of recursive process."""
        last_summation = summation
        summation = self.improve_precision(self.i)
        return self.is_good(summation, last_summation)

    def is_good(self, summ, last_summ):
        # if precision < 0.001, over; else, call `integrate_iter` again.
        if abs(summ - last_summ) <= self.accuracy:
            return summ
        else:
            self.i += 1
            return self.integrate_iter(summ)
...
calc = Integration(X0, XN, 0.001)
val = calc.integrate()
scipy_result = scipy.integrate.quad(f, X0, XN, epsabs=0.001)[0]
print 'The integration is: {0:.8f}'.format(val[0])
print 'The error is: {0:.8f}'.format(abs(val[0] - scipy_result))
print 'The number of recursions is: {0:d}'.format(val[1])
print 'The number of intervals is: {0:d}'.format(2**val[1])

我对递归次数不太满意,我不知道它是否应该收敛得那么慢(根据我的经验)。

3 个答案:

答案 0 :(得分:1)

你错过了堆栈跟踪的主线:

RuntimeError: maximum recursion depth exceeded while calling a Python object

您的递归integrate_iter不会终止。

答案 1 :(得分:1)

除非这是一个学习练习,否则你应该使用scipy.integrate.quad - 它以更高级的方式进行自适应整合。

答案 2 :(得分:1)

integrate_iter内需要两个修复程序。

首先,当您增加i时,您不仅需要向i添加1,还需要将i重新定义为i + 1

其次,在您发布的代码中,integrate_iter在执行None块时返回else。这可以防止在integrate内获得改进的结果,在print中需要它,以便可以在def integrate_iter(self, expectation, summation, accuracy): if accuracy > abs(summation - expectation): return summation else: self.i += 1 return self.integrate_iter(expectation, self.improve_precision(self.i), accuracy) 语句中返回。

def integrate_iter(self, expectation, summation, accuracy):
    if accuracy <= abs(summation - expectation):
        self.i += 1
        # Improve summation.
        summation = self.integrate_iter(expectation,
                                        self.improve_precision(self.i), 
                                        accuracy)
    # Summation is good enough; return.
    return summation

虽然上述方法有效,但我认为以这种方式编写更清楚:

REG QUERY HKCU\Environment\