Removing files from data analysis directory

[ou-summer-of-code-2017.git] / 06-tour-shapes / tour-shapes-build-problem-set.ipynb

diff --git a/06-tour-shapes/tour-shapes-build-problem-set.ipynb b/06-tour-shapes/tour-shapes-build-problem-set.ipynb
index 05df3100e5e578478bbb4fa4a1b6cf51670f6d81..5d9c57dd6396809f6d02e2f2d4cc83f3f481cb71 100644 (file)
--- a/06-tour-shapes/tour-shapes-build-problem-set.ipynb
+++ b/06-tour-shapes/tour-shapes-build-problem-set.ipynb
@@ -278,9 +278,14 @@
    },
    "outputs": [],
    "source": [
-    "def plot_trace(trace, colour='k', xybounds=None, fig=None, subplot_details=None, filename=None):\n",
+    "def plot_trace(trace, colour='k', highlight_start=True,\n",
+    "               xybounds=None, fig=None, subplot_details=None, filename=None):\n",
     "    plt.axis('on')\n",
     "    plt.axes().set_aspect('equal')\n",
+    "        \n",
+    "    if highlight_start:\n",
+    "        plt.axes().add_patch(plt.Circle((trace[0].x, trace[0].y), 0.2, color=colour))\n",
+    "        \n",
     "    for s, t in chunks(trace, 2):\n",
     "        w, h = plot_wh[t.dir]\n",
     "        plt.arrow(s.x, s.y, w, h, head_width=0.1, head_length=0.1, fc=colour, ec=colour, length_includes_head=True)\n",