update example to get thumbnail

Author: AntSimi

examples/16_network/pet_group_anim.py

@@ -135,16 +135,19 @@ def update(frame):
 fig = plt.figure(figsize=(16, 9), dpi=50)
 ax = fig.add_axes([0, 0, 1, 1])
 ax.set_aspect("equal"), ax.grid(), ax.set_xlim(26, 34), ax.set_ylim(31, 35.5)
-groups = ax.scatter(
-    e.lon,
-    e.lat,
-    c=NETWORK_GROUPS[0][2],
-    cmap=ListedColormap(["gray", *e.COLORS[:-1]], name="from_list", N=30),
-    vmin=0,
-    vmax=30,
-)
+cmap = ListedColormap(["gray", *e.COLORS[:-1]], name="from_list", N=30)
+kw_s = dict(cmap=cmap, vmin=0, vmax=30)
+groups = ax.scatter(e.lon, e.lat, c=NETWORK_GROUPS[0][2], **kw_s)
 current_contour = ax.plot([], [], "k", lw=2, label="Current contour")[0]
 matched_contour = ax.plot([], [], "r", lw=1, ls="--", label="Candidate contour")[0]
 txt = ax.text(29, 35, "", fontsize=25)
 ax.legend(fontsize=25)
 ani = VideoAnimation(fig, update, frames=len(NETWORK_GROUPS), interval=220)
+
+# %%
+# Final Result
+# -----------
+fig = plt.figure(figsize=(16, 9))
+ax = fig.add_axes([0, 0, 1, 1])
+ax.set_aspect("equal"), ax.grid(), ax.set_xlim(26, 34), ax.set_ylim(31, 35.5)
+_ = ax.scatter(e.lon, e.lat, c=NETWORK_GROUPS[-1][2], **kw_s)

examples/16_network/pet_segmentation_anim.py

@@ -116,3 +116,11 @@ def update(i_frame):
     e.contour_lon_e[INDICES[0]], e.contour_lat_e[INDICES[0]], color=cmap.colors[0]
 )[0]
 ani = VideoAnimation(fig, update, frames=range(1, len(TRACKS), 4), interval=125)
+
+# %%
+# Final Result
+# -----------
+fig = plt.figure(figsize=(16, 9))
+ax = fig.add_axes([0.04, 0.06, 0.94, 0.88], projection="full_axes")
+ax.set_xlim(19, 29), ax.set_ylim(31, 35.5), ax.grid()
+_ = ax.scatter(e.lon, e.lat, c=TRACKS[-1], cmap=cmap, vmin=0, vmax=vmax, s=20)

notebooks/python_module/16_network/pet_group_anim.ipynb

@@ -167,7 +167,25 @@
       },
       "outputs": [],
       "source": [
-        "fig = plt.figure(figsize=(16, 9), dpi=50)\nax = fig.add_axes([0, 0, 1, 1])\nax.set_aspect(\"equal\"), ax.grid(), ax.set_xlim(26, 34), ax.set_ylim(31, 35.5)\ngroups = ax.scatter(\n    e.lon,\n    e.lat,\n    c=NETWORK_GROUPS[0][2],\n    cmap=ListedColormap([\"gray\", *e.COLORS[:-1]], name=\"from_list\", N=30),\n    vmin=0,\n    vmax=30,\n)\ncurrent_contour = ax.plot([], [], \"k\", lw=2, label=\"Current contour\")[0]\nmatched_contour = ax.plot([], [], \"r\", lw=1, ls=\"--\", label=\"Candidate contour\")[0]\ntxt = ax.text(29, 35, \"\", fontsize=25)\nax.legend(fontsize=25)\nani = VideoAnimation(fig, update, frames=len(NETWORK_GROUPS), interval=220)"
+        "fig = plt.figure(figsize=(16, 9), dpi=50)\nax = fig.add_axes([0, 0, 1, 1])\nax.set_aspect(\"equal\"), ax.grid(), ax.set_xlim(26, 34), ax.set_ylim(31, 35.5)\ncmap = ListedColormap([\"gray\", *e.COLORS[:-1]], name=\"from_list\", N=30)\nkw_s = dict(cmap=cmap, vmin=0, vmax=30)\ngroups = ax.scatter(e.lon, e.lat, c=NETWORK_GROUPS[0][2], **kw_s)\ncurrent_contour = ax.plot([], [], \"k\", lw=2, label=\"Current contour\")[0]\nmatched_contour = ax.plot([], [], \"r\", lw=1, ls=\"--\", label=\"Candidate contour\")[0]\ntxt = ax.text(29, 35, \"\", fontsize=25)\nax.legend(fontsize=25)\nani = VideoAnimation(fig, update, frames=len(NETWORK_GROUPS), interval=220)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "## Final Result\n\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "collapsed": false
+      },
+      "outputs": [],
+      "source": [
+        "fig = plt.figure(figsize=(16, 9))\nax = fig.add_axes([0, 0, 1, 1])\nax.set_aspect(\"equal\"), ax.grid(), ax.set_xlim(26, 34), ax.set_ylim(31, 35.5)\n_ = ax.scatter(e.lon, e.lat, c=NETWORK_GROUPS[-1][2], **kw_s)"
       ]
     }
   ],

notebooks/python_module/16_network/pet_segmentation_anim.ipynb

@@ -111,6 +111,24 @@
       "source": [
         "def update(i_frame):\n    tr = TRACKS[i_frame]\n    mappable_tracks.set_array(tr)\n    s = 40 * ones(tr.shape)\n    s[tr == 0] = 4\n    mappable_tracks.set_sizes(s)\n\n    indices_frames = INDICES[i_frame]\n    mappable_CONTOUR.set_data(\n        e.contour_lon_e[indices_frames],\n        e.contour_lat_e[indices_frames],\n    )\n    mappable_CONTOUR.set_color(cmap.colors[tr[indices_frames] % len(cmap.colors)])\n    return (mappable_tracks,)\n\n\nfig = plt.figure(figsize=(16, 9), dpi=60)\nax = fig.add_axes([0.04, 0.06, 0.94, 0.88], projection=\"full_axes\")\nax.set_title(f\"{len(e)} observations to segment\")\nax.set_xlim(19, 29), ax.set_ylim(31, 35.5), ax.grid()\nvmax = TRACKS[-1].max()\ncmap = ListedColormap([\"gray\", *e.COLORS[:-1]], name=\"from_list\", N=vmax)\nmappable_tracks = ax.scatter(\n    e.lon, e.lat, c=TRACKS[0], cmap=cmap, vmin=0, vmax=vmax, s=20\n)\nmappable_CONTOUR = ax.plot(\n    e.contour_lon_e[INDICES[0]], e.contour_lat_e[INDICES[0]], color=cmap.colors[0]\n)[0]\nani = VideoAnimation(fig, update, frames=range(1, len(TRACKS), 4), interval=125)"
       ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "## Final Result\n\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "collapsed": false
+      },
+      "outputs": [],
+      "source": [
+        "fig = plt.figure(figsize=(16, 9))\nax = fig.add_axes([0.04, 0.06, 0.94, 0.88], projection=\"full_axes\")\nax.set_xlim(19, 29), ax.set_ylim(31, 35.5), ax.grid()\n_ = ax.scatter(e.lon, e.lat, c=TRACKS[-1], cmap=cmap, vmin=0, vmax=vmax, s=20)"
+      ]
     }
   ],
   "metadata": {