python - 使用for循环的辛普森法则(数字积分)

我正在尝试使用for循环在python中编写Simpson规则，但不断收到断言错误，无法找到原因。

def integrate_numeric(xmin, xmax, N):
    xsum = 0
    msum = 0
    h = (xmax-xmin)//N

    for i in range(0, N):
        xsum += f(xmin + i*h)
        print (xsum)

    for i in range(0,N-1):
        msum += f(xmin + (h/2) + i*h)
        print (msum)

    I = (h/6) * (f(xmin) + 4*(msum) + 2*(xsum) + f(xmax))
    return I

F：

def f(x):
    return (x**2) * numpy.sin(x)

样品：

assert numpy.isclose(integrate_numeric(xmin=0, xmax=4, N=50), 1.096591)

最佳答案

这是代码的固定版本：

import numpy

def integrate_numeric(xmin, xmax, N):
    '''
    Numerical integral of f from xmin to xmax using Simpson's rule with
        N panels.
    '''
    xsum = 0
    msum = 0
    h = (xmax-xmin)/N

    for i in range(1, N):
        xsum += f(xmin + i*h)
        print(xsum)

    for i in range(0, N):
        msum += f(xmin + (h/2) + i*h)
        print(msum)

    I = (h/6) * (f(xmin) + 4*msum + 2*xsum + f(xmax))
    return I


def f(x):
    '''Function equivalent to x^2 sin(x).'''
    return (x**2) * numpy.sin(x)


assert numpy.isclose(integrate_numeric(xmin=0, xmax=4, N=50), 1.096591)

笔记：


两个for循环中的范围已更改：我们希望第一个for循环以xmin + h步长（因此xmin + (N-1)*h总值）从h到N-1，第二个为循环以xmin + h/2（xmin + (N-1)*h + h/2总值）的步长从h转到N。
在最终计算中，无需将f应用于msum和xsum：这些值已经是f值的总和。我们仍然需要评估f的唯一位置是xmin和xmax。 （注意：此问题已在问题编辑中得到解决。）
行h = (xmax-xmin)//N必须为h = (xmax-xmin)/N。您只想在这里进行常规划分，而不是楼层划分。这可能是您最初得到零的原因：h应该是0。