add heatmap

Author: malmans2

notebooks/wp5/xch4_lev2_growth_rates.ipynb

@@ -94,17 +94,17 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "def weight_dataset(ds):\n",
-    "    weights = np.abs(np.cos(np.deg2rad(ds[\"latitude\"])))\n",
-    "    return ds.weighted(weights)\n",
+    "def weight_dataset(obj):\n",
+    "    weights = np.abs(np.cos(np.deg2rad(obj[\"latitude\"])))\n",
+    "    return obj.weighted(weights)\n",
     "\n",
     "\n",
-    "def spatial_weighted_mean(ds, dim=None):\n",
-    "    return weight_dataset(ds).mean(dim, keep_attrs=True)\n",
+    "def spatial_weighted_mean(obj, dim=None):\n",
+    "    return weight_dataset(obj).mean(dim, keep_attrs=True)\n",
     "\n",
     "\n",
-    "def spatial_weighted_std(ds, dim=None):\n",
-    "    return weight_dataset(ds).std(dim, keep_attrs=True)\n",
+    "def spatial_weighted_std(obj, dim=None):\n",
+    "    return weight_dataset(obj).std(dim, keep_attrs=True)\n",
     "\n",
     "\n",
     "def regrid(ds, d_lon, d_lat, lon1=180):\n",
@@ -240,7 +240,7 @@
    "id": "15",
    "metadata": {},
    "source": [
-    "## Plot yearly mean growth rate"
+    "## Heat map"
    ]
   },
   {
@@ -249,6 +249,30 @@
    "id": "16",
    "metadata": {},
    "outputs": [],
+   "source": [
+    "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",
+    "for ax in facet.axs.flatten():\n",
+    "    for label in ax.get_xticklabels():\n",
+    "        label.set_rotation(90)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "17",
+   "metadata": {},
+   "source": [
+    "## Plot yearly mean growth rate"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "18",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "grouped = da.groupby(\"time.year\")\n",
     "da_mean = grouped.map(spatial_weighted_mean, dim=[\"time\", \"latitude\"])\n",