50 / 30 : set arguments for exact plotter

Clean ipynb 8bfa77e

Author: kangwonlee

50_ode/30_Runge_Kutta.ipynb

@@ -270,13 +270,11 @@
    ]
   },
   {
-   "cell_type": "code",
-   "execution_count": null,
+   "cell_type": "markdown",
    "metadata": {},
-   "outputs": [],
    "source": [
-    "def exact(t):\n",
-    "    return x_0 * py.exp((-a_1 / a_0) * t)\n",
+    "Initial value $x(t_0)$<br>초기값 $x(t_0)$\n",
+    "\n",
     "\n"
    ]
   },
@@ -286,15 +284,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import ode_solver\n",
+    "x_0 = 4.5\n",
     "\n"
    ]
   },
   {
-   "cell_type": "markdown",
+   "cell_type": "code",
+   "execution_count": null,
    "metadata": {},
+   "outputs": [],
    "source": [
-    "$\\Delta t$\n",
+    "def exact(t):\n",
+    "    return x_0 * py.exp((-a_1 / a_0) * t)\n",
     "\n"
    ]
   },
@@ -304,17 +305,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "delta_t = 1.0\n",
-    "t_sec_array = np.arange(0, 6 + delta_t*0.5, delta_t)\n",
+    "import ode_solver\n",
     "\n"
    ]
   },
   {
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "초기값<br>Initial value<br>\n",
-    "$x(t_0)$\n",
+    "$\\Delta t$\n",
     "\n"
    ]
   },
@@ -324,7 +323,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "x_0 = 4.5\n",
+    "delta_t = 0.01\n",
+    "t_sec_array = np.arange(0, 6 + delta_t*0.5, delta_t)\n",
     "\n"
    ]
   },
@@ -419,7 +419,10 @@
     "\n",
     "# *** Exact Solution\n",
     "t_exact_array = np.linspace(0, 6)\n",
-    "exact = ode_plot.ExactPlotterFirstOrderODE(t_exact_array)\n",
+    "exact = ode_plot.ExactPlotterFirstOrderODE(\n",
+    "    t_exact_array,\n",
+    "    a_0=a_0, a_1=a_1, x_0=x_0)\n",
+    "\n",
     "exact.plot()\n",
     "\n",
     "py.xlabel('t(sec)')\n",
@@ -481,8 +484,11 @@
     "py.plot(t_rk4___out, x_rk4___out, '.-', label='RK4')\n",
     "\n",
     "# plot exact solution\n",
-    "exact = ode_plot.ExactPlotterFirstOrderODE(py.array(t_rk4___out))\n",
+    "exact = ode_plot.ExactPlotterFirstOrderODE(\n",
+    "    py.array(t_rk4___out),\n",
+    "    a_0=a_0, a_1=a_1, x_0=x_0)\n",
     "exact.plot()\n",
+    "\n",
     "py.grid(True)\n",
     "py.xlabel('t(sec)')\n",
     "py.ylabel('y(t)')\n",