update example

Author: AntSimi

examples/02_eddy_identification/pet_eddy_detection.py

@@ -4,6 +4,8 @@
 
 Script will detect eddies on adt field, and compute u,v with method add_uv(which could use, only if equator is avoid)
 
+Figures will show different step to detect eddies.
+
 """
 from datetime import datetime
 from matplotlib import pyplot as plt
@@ -12,10 +14,9 @@
 
 
 def start_axes(title):
-    fig = plt.figure(figsize=(12.5, 5))
-    ax = fig.add_axes([0.03, 0.03, 0.94, 0.94])
-    ax.set_xlim(-5, 37)
-    ax.set_ylim(30, 46)
+    fig = plt.figure(figsize=(13, 5))
+    ax = fig.add_axes([0.03, 0.03, 0.90, 0.94])
+    ax.set_xlim(-6, 36.5), ax.set_ylim(30, 46)
     ax.set_aspect("equal")
     ax.set_title(title)
     return ax
@@ -31,14 +32,21 @@ def update_axes(ax, mappable=None):
     data.get_path("dt_med_allsat_phy_l4_20160515_20190101.nc"), "longitude", "latitude"
 )
 
-ax = start_axes("ADT(m)")
+ax = start_axes("ADT (m)")
 m = g.display(ax, "adt", vmin=-0.15, vmax=0.15)
 update_axes(ax, m)
 
 g.add_uv("adt")
+ax = start_axes("U/V deduce from ADT (m)")
+ax.set_xlim(2.5, 9), ax.set_ylim(37.5, 40)
+m = g.display(ax, "adt", vmin=-0.15, vmax=0.15)
+u, v = g.grid("u").T, g.grid("v").T
+ax.quiver(g.x_c, g.y_c, u, v, scale=10)
+update_axes(ax, m)
+
 g.bessel_high_filter("adt", 500, order=2)
 
-ax = start_axes("ADT (m) filtered(500km, order 2)")
+ax = start_axes("ADT (m) filtered (500km, order 2)")
 m = g.display(ax, "adt", vmin=-0.15, vmax=0.15)
 update_axes(ax, m)
 
@@ -60,19 +68,21 @@ def update_axes(ax, mappable=None):
 ax = start_axes("ADT contour reject but which contain eddies")
 g.contours.label_contour_unused_which_contain_eddies(a)
 g.contours.label_contour_unused_which_contain_eddies(c)
-g.contours.display(ax, only_contain_eddies=True, color="k", lw=1, label="Could be interaction contour")
-a.display(ax, color="b", linewidth=0.5, label="Anticyclonic", ref=-10)
-c.display(ax, color="r", linewidth=0.5, label="Cyclonic", ref=-10)
+g.contours.display(
+    ax, only_contain_eddies=True, color="k", lw=1, label="Could be interaction contour"
+)
+a.display(ax, color="r", linewidth=0.5, label="Anticyclonic", ref=-10)
+c.display(ax, color="b", linewidth=0.5, label="Cyclonic", ref=-10)
 ax.legend()
 update_axes(ax)
 
 ax = start_axes("Eddies detected")
-a.display(ax, color="b", linewidth=0.5, label="Anticyclonic", ref=-10)
-c.display(ax, color="r", linewidth=0.5, label="Cyclonic", ref=-10)
+a.display(ax, color="r", linewidth=0.5, label="Anticyclonic", ref=-10)
+c.display(ax, color="b", linewidth=0.5, label="Cyclonic", ref=-10)
 ax.legend()
 update_axes(ax)
 
 ax = start_axes("Eddies speed radius (km)")
-a.scatter(ax, "radius_s", vmin=10, vmax=50, s=80, ref=-10, cmap='jet', factor=0.001)
-m = c.scatter(ax, "radius_s", vmin=10, vmax=50, s=80, ref=-10, cmap='jet', factor=0.001)
+a.scatter(ax, "radius_s", vmin=10, vmax=50, s=80, ref=-10, cmap="jet", factor=0.001)
+m = c.scatter(ax, "radius_s", vmin=10, vmax=50, s=80, ref=-10, cmap="jet", factor=0.001)
 update_axes(ax, m)

examples/08_tracking_manipulation/pet_display_field.py

@@ -18,8 +18,7 @@
 fig = plt.figure(figsize=(12, 6))
 ax = fig.add_axes((0.05, 0.1, 0.9, 0.9))
 ax.set_aspect("equal")
-ax.set_xlim(-5, 37)
-ax.set_ylim(30, 46)
+ax.set_xlim(-6, 36.5), ax.set_ylim(30, 46)
 m = c.scatter(ax, "amplitude", ref=-10, vmin=0, vmax=0.1)
 ax.grid()
 

examples/08_tracking_manipulation/pet_display_track.py

@@ -24,8 +24,7 @@
 fig = plt.figure(figsize=(12, 5))
 ax = fig.add_axes((0.05, 0.1, 0.9, 0.9))
 ax.set_aspect("equal")
-ax.set_xlim(-5, 37)
-ax.set_ylim(30, 46)
+ax.set_xlim(-6, 36.5), ax.set_ylim(30, 46)
 a.plot(ax, ref=-10, label="Anticyclonic", color="r", lw=0.1)
 c.plot(ax, ref=-10, label="Cyclonic", color="b", lw=0.1)
 ax.legend()

examples/10_tracking_diagnostics/pet_center_count.py

@@ -17,14 +17,13 @@
 )
 a = a.merge(c)
 # Plot
-fig = plt.figure(figsize=(20, 8))
-ax = fig.add_subplot(111)
-ax.set_xlim(-5, 37)
-ax.set_ylim(30, 46)
+fig = plt.figure(figsize=(13.5, 5))
+ax = fig.add_axes([0.03, 0.03, 0.90, 0.94])
+ax.set_xlim(-6, 36.5), ax.set_ylim(30, 46)
 ax.set_aspect("equal")
 step = 0.1
 t0, t1 = a.period
-g = a.grid_count(((-6, 37, step), (30, 46, step)), center=True)
+g = a.grid_count(((-7, 37, step), (30, 46, step)), center=True)
 m = g.display(ax, name="count", vmin=0, vmax=2, factor=1 / (step ** 2 * (t1 - t0)))
 ax.grid()
 cb = plt.colorbar(m, cax=fig.add_axes([0.95, 0.05, 0.01, 0.9]))

examples/10_tracking_diagnostics/pet_geographic_stats.py

@@ -9,6 +9,15 @@
 import py_eddy_tracker_sample
 
 
+def start_axes(title):
+    fig = plt.figure(figsize=(13.5, 5))
+    ax = fig.add_axes([0.03, 0.03, 0.90, 0.94])
+    ax.set_xlim(-6, 36.5), ax.set_ylim(30, 46)
+    ax.set_aspect("equal")
+    ax.set_title(title)
+    return ax
+
+
 a = TrackEddiesObservations.load_file(
     py_eddy_tracker_sample.get_path("eddies_med_adt_allsat_dt2018/Anticyclonic.zarr")
 )
@@ -17,42 +26,27 @@
 )
 a = a.merge(c)
 
-fig = plt.figure(figsize=(15, 7))
-ax = fig.add_subplot(111)
-ax.set_xlim(-5, 37)
-ax.set_ylim(30, 46)
-ax.set_aspect("equal")
 step = 0.1
+ax = start_axes("Amplitude mean by box of %s°" % step)
 g = a.grid_stat(((-5, 37, step), (30, 46, step)), "amplitude")
-m = g.display(ax, name="amplitude", vmin=0, vmax=0.2)
+m = g.display(ax, name="amplitude", vmin=0, vmax=10, factor=100)
 ax.grid()
-cb = plt.colorbar(m, cax=fig.add_axes([0.92, 0.05, 0.01, 0.9]))
-cb.set_label("Amplitude (m)")
-ax.set_title("Amplitude mean by box of %s°" % step)
-
-fig = plt.figure(figsize=(15, 7))
-ax = fig.add_subplot(111)
-ax.set_xlim(-5, 37)
-ax.set_ylim(30, 46)
-ax.set_aspect("equal")
+cb = plt.colorbar(m, cax=ax.figure.add_axes([0.94, 0.05, 0.01, 0.9]))
+cb.set_label("Amplitude (cm)")
+
+ax = start_axes("Speed radius mean by box of %s°" % step)
 step = 0.1
-g = a.grid_stat(((-5, 37, step), (30, 46, step)), "radius_s")
+g = a.grid_stat(((-7, 37, step), (30, 46, step)), "radius_s")
 m = g.display(ax, name="radius_s", vmin=10, vmax=50, factor=0.001)
 ax.grid()
-cb = plt.colorbar(m, cax=fig.add_axes([0.92, 0.05, 0.01, 0.9]))
+cb = plt.colorbar(m, cax=ax.figure.add_axes([0.94, 0.05, 0.01, 0.9]))
 cb.set_label("Speed radius (km)")
-ax.set_title("Speed radius mean by box of %s°" % step)
 
-fig = plt.figure(figsize=(15, 7))
-ax = fig.add_subplot(111)
-ax.set_xlim(-5, 37)
-ax.set_ylim(30, 46)
-ax.set_aspect("equal")
+ax = start_axes("Percent of virtual by box of %s°" % step)
 step = 0.1
-g = a.grid_stat(((-5, 37, step), (30, 46, step)), "virtual")
+g = a.grid_stat(((-7, 37, step), (30, 46, step)), "virtual")
 g.vars["virtual"] *= 100
 m = g.display(ax, name="virtual", vmin=0, vmax=15)
 ax.grid()
-cb = plt.colorbar(m, cax=fig.add_axes([0.92, 0.05, 0.01, 0.9]))
+cb = plt.colorbar(m, cax=ax.figure.add_axes([0.94, 0.05, 0.01, 0.9]))
 cb.set_label("Percent of virtual (%)")
-ax.set_title("Percent of virtual by box of %s°" % step)

examples/10_tracking_diagnostics/pet_pixel_used.py

@@ -18,12 +18,12 @@
 t0, t1 = a.period
 
 # Plot
-fig = plt.figure(figsize=(15, 20))
-ax_a = fig.add_subplot(311)
+fig = plt.figure(figsize=(12, 14))
+ax_a = fig.add_axes([0.03, 0.66, 0.90, 0.33])
 ax_a.set_title("Anticyclonic frequency")
-ax_c = fig.add_subplot(312)
+ax_c = fig.add_axes([0.03, 0.33, 0.90, 0.33])
 ax_c.set_title("Cyclonic frequency")
-ax_all = fig.add_subplot(313)
+ax_all = fig.add_axes([0.03, 0, 0.90, 0.33])
 ax_all.set_title("All eddies frequency")
 
 step = 0.1
@@ -45,8 +45,7 @@
 
 for ax in (ax_a, ax_c, ax_all):
     ax.set_aspect("equal")
-    ax.set_xlim(-5, 37)
-    ax.set_ylim(30, 46)
+    ax.set_xlim(-6, 36.5), ax.set_ylim(30, 46)
     ax.grid()
 
 plt.colorbar(m, cax=fig.add_axes([0.95, 0.05, 0.01, 0.9]))