update labels

Author: malmans2

notebooks/wp5/xch4_lev2_growth_rates.ipynb

@@ -163,7 +163,14 @@
     "        .squeeze()\n",
     "    )\n",
     "    da.attrs = {\"units\": \"ppm/year\", \"long_name\": \"Growth Rate\"}\n",
-    "    return da"
+    "    return da\n",
+    "\n",
+    "\n",
+    "def band_from_central_latitude(latitude):\n",
+    "    latitudes = [latitude - 10, latitude + 10]\n",
+    "    return \" – \".join(\n",
+    "        [f\"{abs(int(lat))}°{'S' if lat < 0 else 'N'}\" for lat in latitudes]\n",
+    "    )"
    ]
   },
   {
@@ -229,8 +236,9 @@
    "outputs": [],
    "source": [
     "facet = da.plot(col=\"latitude\", col_wrap=3, hue=\"method\")\n",
-    "for ax in facet.axs.flatten():\n",
+    "for ax, sel_dict in zip(facet.axs.flatten(), facet.name_dicts.flatten()):\n",
     "    ax.grid()\n",
+    "    ax.set_title(band_from_central_latitude(sel_dict[\"latitude\"]))\n",
     "    for label in ax.get_xticklabels():\n",
     "        label.set_rotation(90)"
    ]
@@ -253,9 +261,13 @@
     "with xr.set_options(keep_attrs=True):\n",
     "    anomaly = da - spatial_weighted_mean(da)\n",
     "anomaly.attrs[\"long_name\"] = \"Δ\" + anomaly.long_name\n",
-    "facet = anomaly.plot(row=\"method\", robust=True, x=\"time\", figsize=(10, 10))\n",
+    "yticks = anomaly[\"latitude\"].values.tolist()\n",
+    "facet = anomaly.plot(\n",
+    "    row=\"method\", robust=True, x=\"time\", figsize=(10, 10), yticks=yticks\n",
+    ")\n",
     "for ax in facet.axs.flatten():\n",
-    "    ax.grid()"
+    "    ax.grid()\n",
+    "    ax.set_yticklabels([band_from_central_latitude(lat) for lat in ax.get_yticks()])"
    ]
   },
   {