方法 Euler TypeError: cannot unpack non-iterable NoneType object

Question

尝试方法 Euler 和 Runge-kutta，但收到错误：

line 59, in <module>
    xloc , yloc, h , er = rkf45step(xrk[-1],yrk[-1],h,f)
TypeError: cannot unpack non-iterable NoneType object"

为什么我收到此错误？ 你能帮我吗。 我尝试了更大的超时，但无法解决此问题。

import numpy as np
import matplotlib.pyplot as plt#
beta = -10.
y0 = 1.
h = 0.02
xL = 0.
xR = 1.
N = int((xR-xL)/h)
h = (xR-xL)/float(N)
x = np.linspace(xL,xR,N)
y = np.zeros(N)
xexact = np.linspace(xL,xR,max(1000.,100*N))
def f(x,y):
    return beta * y
def exact(x):
    return y0*np.exp(beta*x)
def eulerIncrementFunction(x,yn,h,ode):
    return ode(x,yn)
def rkf45step(x,yn,h,ode):
    hmin = 1e-5
    hmax = 5e-1
    emin = 1e-7
    nMax = 100
    emax = 1e-5
    if x+h > xR:
        h = xR-x
        update = 0
        for i in range(nMax):
            k1 = ode(x,yn)
            k2 = ode(x+h/4.,yn+h/4.*k1)
            k3 = ode(x+3./8.*h,yn+3./32.*h*k1-9./32.*h*k2)
            k4 = ode(x+12./13.*h,yn+1932./2197.*h*k1-7200./2197.*h*k2+7296./2197.*h*k3)
            k5 = ode(x+h,yn+439./216.*h*k1-8.*h*k2+3680./513.*h*k3-845./4104.*h*k4)
            k6 = ode(x+h/2.,yn-8./27.*h*k1+2.*h*k2-3544./2565.*h*k3+1859./4140.*h*k4-11./40.*h*k5)
            y4 = yn + h * (25./216*k1 + 1408./2565.*k3+2197./4104.*k4-1./5.*k5) 
            y5 = yn + h * (16./135.*k1 + 6656./12825.*k3 + 28561./56430.*k4 - 9./50.*k5 +2./55.*k6)
            er = np.abs(y5-y4)
            if er < emin:
                h = min(2.*h,hmax)
                if x+h > xR:
                    h = xR-x
                    break
                elif er > emax:
                        h = max(h/2.,hmin)
                else:
                    break
            if i==nMax-1:
                print ("max number of iterations reached, check parameters")
            return x+h, y5, h , er
y[0] = y0
for i in range(N-1):
    y[i+1] = y[i] + h * eulerIncrementFunction(x[i],y[i],h,f)
nMax = 1000
xrk = np.zeros(1)
yrk = y0*np.ones(1)
hrk = np.zeros(1)
h = 0.5
for i in range(nMax):
    xloc , yloc, h , er = rkf45step(xrk[-1],yrk[-1],h,f)
    xrk = np.append(xrk,xloc)
    yrk = np.append(yrk,yloc)
    if i==0:
        hrk[i] = h
    else:
        hrk = np.append(hrk,h)
    if xrk[-1]==xR:
        break
plt.subplot(211)
plt.plot(xexact,exact(xexact))
plt.plot(x,y)
plt.plot(xrk,yrk, markersize=7,markeredgewidth=1)
plt.legend()

Answer 1

在这条线上：

xloc , yloc, h , er = rkf45step(xrk[-1],yrk[-1],h,f)

xrk[-1]将为 0， h为 0.5。

现在在你的 function, rkf45step中，你有这个检查：

if x+h > xR:

x+h的计算结果为 0.5， xR为 1，因此该条件的计算结果为False 。 因此，此条件中的任何代码都不会被执行，包括您的唯一一个return语句。 在没有执行返回的情况下，默认情况下 function 将返回None 。 因此，您尝试将单个None解压缩为 4 个变量（ xloc, yloc, h, er ）。 这是不可能的，因此您看到的是例外。

解决方案是修复rkf45step function 中的逻辑以始终返回 4 个值。 或者修复对 function 的调用以先检查None ：

values = rkf45step(xrk[-1],yrk[-1],h,f)
if values is None:
    # take some corrective action here knowing that values is None
else:
    xloc , yloc, h , er = values