Question

编辑：函数为u [t] = [[0，1]，[-6,5]] * u，初始条件为u（0）= [[1]，[1]]。确切的答案是u_t = [[1]，[2]] * 2 * e **（2 * t）-[[1]，[3]] * e **（3 * t）

我正在从事一个项目，但是我为什么不起作用却迷失了方向。我打算在以下代码上运行Euler方法方案。

import numpy as np
from matplotlib import pyplot as plt

def Eulm():

    x0=0
    y0=1
    z0=1
    n=21
    xf=2
    y0=1
    z0=1
    w0=y0
    q0=z0
    w = [0] * (n+1)
    q = [0] * (n+1)
    w[0]=w0
    q[0]=q0
    x=np.linspace(x0,xf,n)
    y=2*np.e**(2*x)-np.e**(3*x)
    z=4*np.e**(2*x)-3*np.e**(3*x)
    L=[0]

    for i in range (1,n):
        deltax=(xf-x0)/(n-1)
        y0=y[0]
        z0=z[0]
        A=np.matrix([[1, -deltax], [6*deltax, 1-5*deltax]])
        G=np.linalg.inv(A)
        print(G)
        b=np.matrix([y[i-1], z[i-1]])
        b=b.transpose
        w[i]=G[0][0]*w[0][i]+G[0][1]*q[0][i-1]
        q[i]=G[1][0]*w[0][i-1]+G[1][1]*q[0][i-1]
        plt.plot(x,y)
        plt.xlabel('Time')
        plt.ylabel('Numerical Solutions')
        plt.title('Numerical Solutions with respect to Time')
        plt.show()

我遇到了以下错误，并且由于要遍历列表，因此不确定为什么会出现此错误。

TypeError                                 Traceback (most recent call last)
<ipython-input-49-da43a543ec92> in <module>()
----> 1 Eulm()

    <ipython-input-47-022f3e9099b9> in Eulm()
         29         b=np.matrix([y[i-1], z[i-1]])
         30         b=b.transpose
    ---> 31         w[i]=G[0][0]*w[0][i]+G[0][1]*q[0][i-1]
         32         q[i]=G[1][0]*w[0][i-1]+G[1][1]*q[0][i-1]
         33         plt.plot(x,y)

TypeError: 'int' object is not subscriptable

任何帮助将不胜感激。谢谢！

Answer 1

您的代码/ Euler方法的理论实现存在很多问题。这是eulm的简单版本，可完成预期的工作：

import numpy as np
from matplotlib import pyplot as plt

def eulm(n=2001):
    x0=0
    y0=1
    z0=1
    xf=2
    wq = np.zeros((n, 2))
    wq[0] = y0,z0
    x = np.linspace(x0,xf,n)
    y = 2*np.exp(2*x) - np.exp(3*x)
    z = 4*np.exp(2*x) - 3*np.exp(3*x)
    A = np.array([[0, 1], [-6, 5]])
    dx = (x[1] - x[0])

    fig = plt.figure(figsize=(10,6))
    ax = fig.gca()
    ax.set_xlabel('Time')
    ax.set_ylabel('Numerical Solutions')
    ax.set_title('All Numerical Solutions with respect to Time,\n%d timesteps' % n)

    for i in range (1,n):
        wq[i] = wq[i - 1] + dx*(A @ wq[i - 1])
        if i == n-1:
            # add a legend in the final display
            ax.plot(x, wq[:, 0], c='C0', label='approx y')
            ax.plot(x, wq[:, 1], c='C1', label='approx z')
            ax.legend()
            fig.show()
        else:
            ax.plot(x, wq[:, 0], c='C0')
            ax.plot(x, wq[:, 1], c='C1')
            fig.show()

    fig = plt.figure(figsize=(10,6))
    ax = fig.gca()
    ax.set_xlabel('Time')
    ax.set_ylabel('Numerical Solutions')
    ax.set_title('Final Numerical Solutions with respect to Time,\n%d timesteps' % n)
    ax.plot(x, wq[:, 0], ':', c='k', lw=3, zorder=99, label='approx y')
    ax.plot(x, y, c='C2', lw=15, label='exact y')
    ax.plot(x, wq[:, 1], '--', c='k', lw=3, zorder=99, label='approx z')
    ax.plot(x, z, c='C3', lw=15, label='exact z')
    ax.legend()
    fig.show()

    return wq, x, y, z

我将w和q卷成一个两列的数组wq。我摆脱了逆矩阵G（我无法弄清楚您要如何处理它，无论如何它都不是standard Euler method的一部分）。我使用来自Numpy的一些矩阵表示法简化了wq的值的下一个步骤的计算（@运算符最近被添加到Numpy中，并用作矩阵乘法运算符）。

对于足够大的n（以及足够小的时间步长dx），wq[:, 0]和wq[:, 1]中的估计值收敛到{{1 }}和y的错误很小（z时大约~1%）。这是n=2001运行时产生的图，如下所示：

欧拉方法方案python

1 个答案: