{ "metadata": { "name": "", "signature": "sha256:b97929dc33b7d96954adc7d93f08ed16f6ed131d1d67a6141093160a7f2a5897" }, "nbformat": 3, "nbformat_minor": 0, "worksheets": [ { "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "Consider the ODE\n", "$$\n", "y' = y, \\qquad t \\ge 0\n", "$$\n", "with initial condition\n", "$$\n", "y(0) = 1\n", "$$\n", "The exact solution is\n", "$$\n", "y(t) = \\exp(t)\n", "$$" ] }, { "cell_type": "code", "collapsed": false, "input": [ "import numpy as np\n", "from matplotlib import pyplot as plt" ], "language": "python", "metadata": {}, "outputs": [], "prompt_number": 1 }, { "cell_type": "markdown", "metadata": {}, "source": [ "Right hand side function" ] }, { "cell_type": "code", "collapsed": false, "input": [ "def f(t,y):\n", " return y" ], "language": "python", "metadata": {}, "outputs": [], "prompt_number": 2 }, { "cell_type": "markdown", "metadata": {}, "source": [ "Exact solution" ] }, { "cell_type": "code", "collapsed": false, "input": [ "def yexact(t):\n", " return np.exp(t)" ], "language": "python", "metadata": {}, "outputs": [], "prompt_number": 3 }, { "cell_type": "markdown", "metadata": {}, "source": [ "This implements Euler method\n", "$$\n", "y_n = y_{n-1} + h f(t_{n-1},y_{n-1})\n", "$$" ] }, { "cell_type": "code", "collapsed": false, "input": [ "def euler(t0,T,y0,h):\n", " N = int((T-t0)/h)\n", " y = np.zeros(N)\n", " t = np.zeros(N)\n", " y[0] = y0\n", " t[0] = t0\n", " for n in range(1,N):\n", " y[n] = y[n-1] + h*f(t[n-1],y[n-1])\n", " t[n] = t[n-1] + h\n", " return t, y" ], "language": "python", "metadata": {}, "outputs": [], "prompt_number": 4 }, { "cell_type": "code", "collapsed": false, "input": [ "t0 = 0.0\n", "y0 = 1.0\n", "T = 5.0\n", "\n", "te = np.linspace(t0,T,100)\n", "ye = yexact(te)\n", "plt.plot(te,ye,'--')\n", "\n", "h = 0.2\n", "t,y = euler(t0,T,y0,h)\n", "plt.plot(t,y)\n", "\n", "h = 0.1\n", "t,y = euler(t0,T,y0,h)\n", "plt.plot(t,y)\n", "\n", "h = 0.05\n", "t,y = euler(t0,T,y0,h)\n", "plt.plot(t,y)\n", "\n", "plt.legend(('Exact','0.2','0.1','0.05'),loc=2)\n", "plt.xlabel('t')\n", "plt.ylabel('y')" ], "language": "python", "metadata": {}, "outputs": [ { "metadata": {}, "output_type": "pyout", "prompt_number": 5, "text": [ "" ] }, { "metadata": {}, "output_type": "display_data", "svg": [ "\n", "\n", "\n", "\n" ], "text": [ "" ] } ], "prompt_number": 5 }, { "cell_type": "code", "collapsed": false, "input": [], "language": "python", "metadata": {}, "outputs": [], "prompt_number": 5 } ], "metadata": {} } ] }