do not save soil moisture drought index as log10

IPCC-WG1 · Feb 9, 2023 · dcb2d95 · dcb2d95
1 parent 33ebf69
commit dcb2d95
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/code/SMDroughtIndex.ipynb b/code/SMDroughtIndex.ipynb
@@ -984,6 +984,11 @@
     "\n",
     "# save figure data\n",
     "# =================================================================\n",
+    "# NOTE: use original data not in log10\n",
+    "# the order of operations (log10 and median leads to a\n",
+    "# difference in a few grid points)\n",
+    "\n",
+    "dta_ = seas_drought_freq_wrt_clim\n",
     "\n",
     "varn = \"SM_drought_index\"\n",
     "figure = \"Figure 11.19\"\n",
@@ -1005,10 +1010,10 @@
     "        add_prefix=False,\n",
     "    )\n",
     "\n",
-    "    da = dta_[i]\n",
+    "    da = dta_[str(warming_level)].sel(season=season)\n",
     "\n",
     "    da.attrs[\"long_name\"] = \"Total Soil Moisture Drought\"\n",
-    "    da.attrs[\"comment\"] = \"anomaly wrt 1850-1900\"\n",
+    "    da.attrs[\"comment\"] = \"anomaly wrt 1850-1900; best shown on a log-scale\"\n",
     "\n",
     "    ds = sfd.map_panel(\n",
     "        da=da,\n",
@@ -1101,6 +1106,11 @@
     "\n",
     "# save figure data\n",
     "# =================================================================\n",
+    "# NOTE: use original data not in log10\n",
+    "# the order of operations (log10 and median leads to a\n",
+    "# difference in a few grid points)\n",
+    "\n",
+    "dta_ = seas_drought_freq_wrt_clim\n",
     "\n",
     "varn = \"SM_drought_index\"\n",
     "figure = \"Figure 11.19\"\n",
@@ -1122,10 +1132,11 @@
     "        add_prefix=False,\n",
     "    )\n",
     "    \n",
-    "    da=dta_[i]\n",
+    "    da = dta_[str(warming_level)].sel(season=season)\n",
+    "\n",
     "    \n",
     "    da.attrs[\"long_name\"] = \"Total Soil Moisture Drought\"\n",
-    "    da.attrs[\"comment\"] = \"anomaly wrt 1850-1900\"\n",
+    "    da.attrs[\"comment\"] = \"anomaly wrt 1850-1900; best shown on a log-scale\"\n",
     "\n",
     "    ds = sfd.map_panel(\n",
     "        da=da,\n",