Ошибка метода Euler TypeError: невозможно распаковать не повторяющийся объект NoneType

Question

Пробуем метод Эйлера и Рунге-кутта, но получаю ошибку:

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.

Теперь в вашей функции rkf45step у вас есть такая проверка:

if x+h > xR:

x+h оценивается как 0,5, а xR равно 1, поэтому это условие оценивается как False. Следовательно, ни один из кодов внутри этого условия не выполняется, включая ваш единственный оператор return. В отсутствие выполненного возврата функция по умолчанию возвращает None. Таким образом, вы пытаетесь распаковать один None в 4 переменные (xloc, yloc, h, er). Это невозможно и, следовательно, исключение, которое вы видите.

Решение состоит в том, чтобы исправить либо logi c в вашей функции rkf45step, чтобы всегда возвращать 4 значения. Или исправьте вызов функции, чтобы сначала проверить 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