DOC: add section about Xarray datasets

Author: mgeier

doc/code-cells.ipynb

@@ -478,6 +478,49 @@
     "df"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Xarray Datasets\n",
+    "\n",
+    "[Xarray datasets](https://docs.xarray.dev/en/stable/generated/xarray.Dataset.html)\n",
+    "should also be shown in their beautiful HTML representation.\n",
+    "LaTeX/PDF output will fall back to the plain text representation."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import xarray as xr"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "a = {}\n",
+    "for name in [\"t\", \"q\"]: \n",
+    "    da = xr.DataArray(\n",
+    "        [\n",
+    "            [[11, 12, 13], [21, 22, 23], [31, 32, 33]],\n",
+    "            [[14, 15, 16], [24, 25, 26], [34, 35, 36]],\n",
+    "        ],\n",
+    "        coords={\"level\": [\"500\", \"700\"], \"x\": [1, 2, 3], \"y\": [4, 5, 6]},\n",
+    "        name=name,\n",
+    "        attrs={\"param\": name}\n",
+    "    )\n",
+    "    a[name] = da\n",
+    "\n",
+    "ds = xr.Dataset(a, attrs={\"title\": \"Demo dataset\", \"year\": 2024})\n",
+    "ds"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},

doc/requirements.txt

@@ -3,6 +3,7 @@ numpy
 matplotlib
 ipympl
 pandas
+xarray
 # setuptools seems to be needed for pybtex, see
 # https://github.com/mcmtroffaes/sphinxcontrib-bibtex/issues/345:
 setuptools