escape hatches and a exercise

Author: keewis

intermediate/hierarchical_computation.ipynb

@@ -235,6 +235,78 @@
     "\n",
     "tree.map_over_datasets(demean)"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "21",
+   "metadata": {},
+   "source": [
+    "## Escape hatches\n",
+    "\n",
+    "For some more complex operations, it might make sense to work on {py:class}`xarray.Dataset` or {py:class}`xarray.DataArray` objects and reassemble the tree afterwards.\n",
+    "\n",
+    "Let's look at a new dataset:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "22",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "precipitation = xr.tutorial.open_datatree(\"precipitation.nc4\").load()\n",
+    "precipitation"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "23",
+   "metadata": {},
+   "source": [
+    "Suppose we wanted to interpolate the observed precipitation to the modelled precipitation. We could use `map_over_datasets` for this, but we can also have a bit more control:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "24",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "interpolated = xr.DataTree.from_dict(\n",
+    "    {\n",
+    "        \"/\": precipitation.ds,\n",
+    "        \"/observed\": precipitation[\"/observed\"].ds.interp(\n",
+    "            lat=precipitation[\"/reanalysis/lat\"],\n",
+    "            lon=precipitation[\"/reanalysis/lon\"],\n",
+    "        ),\n",
+    "        \"/reanalysis\": precipitation[\"/reanalysis\"],\n",
+    "    }\n",
+    ")\n",
+    "interpolated"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "25",
+   "metadata": {},
+   "source": [
+    "::::{admonition} Exercise\n",
+    ":class: tip\n",
+    "Compute the difference between total observed and modelled precipitation, and plot the result.\n",
+    "\n",
+    ":::{admonition} Solution\n",
+    ":class: dropdown\n",
+    "\n",
+    "```python\n",
+    "total = precipitation.sum(dim=[\"lon\", \"lat\"])\n",
+    "difference = total[\"/observed/precipitation\"] - total[\"/reanalysis/precipitation\"]\n",
+    "difference.plot()\n",
+    "```\n",
+    ":::\n",
+    "::::\n"
+   ]
   }
  ],
  "metadata": {