add regionalisation

Author: malmans2

notebooks/wp5/steric_sea_level_ora5.ipynb

@@ -54,7 +54,7 @@
     "import gsw_xarray as gsw\n",
     "import matplotlib.pyplot as plt\n",
     "import xarray as xr\n",
-    "from c3s_eqc_automatic_quality_control import diagnostics, download\n",
+    "from c3s_eqc_automatic_quality_control import diagnostics, download, utils\n",
     "\n",
     "plt.style.use(\"seaborn-v0_8-notebook\")"
    ]
@@ -92,8 +92,13 @@
    },
    "outputs": [],
    "source": [
+    "# Time\n",
     "year_start = 2004\n",
-    "year_stop = 2023"
+    "year_stop = 2023\n",
+    "\n",
+    "# Space\n",
+    "lat_slice = slice(-60, 60)\n",
+    "lon_slice = slice(-180, 180)"
    ]
   },
   {
@@ -216,10 +221,13 @@
     "    return ssl\n",
     "\n",
     "\n",
-    "def compute_ssl(collection_id, request, prefix, **download_kwargs):\n",
+    "def compute_ssl(\n",
+    "    collection_id, request, prefix, lon_slice, lat_slice, **download_kwargs\n",
+    "):\n",
     "    ds = download.download_and_transform(\n",
     "        collection_id, request, transform_func=compute_gsw_ds, **download_kwargs\n",
     "    )\n",
+    "    ds = utils.regionalise(ds, lon_slice=lon_slice, lat_slice=lat_slice)\n",
     "    if prefix == \"\":\n",
     "        rho = ds[\"rho\"]\n",
     "    elif prefix == \"thermo\":\n",
@@ -232,8 +240,22 @@
     "\n",
     "\n",
     "@cacholote.cacheable\n",
-    "def compute_ssl_timeseries(collection_id, request, prefix, **download_kwargs):\n",
-    "    ssl = compute_ssl(collection_id, request, prefix, **download_kwargs)\n",
+    "def compute_ssl_timeseries(\n",
+    "    collection_id,\n",
+    "    request,\n",
+    "    prefix,\n",
+    "    lon_slice,\n",
+    "    lat_slice,\n",
+    "    **download_kwargs,\n",
+    "):\n",
+    "    ssl = compute_ssl(\n",
+    "        collection_id,\n",
+    "        request,\n",
+    "        prefix,\n",
+    "        lon_slice=lon_slice,\n",
+    "        lat_slice=lat_slice,\n",
+    "        **download_kwargs,\n",
+    "    )\n",
     "    return diagnostics.spatial_weighted_mean(ssl)"
    ]
   },
@@ -273,7 +295,14 @@
     "dataarrays = []\n",
     "for prefix in [\"\", \"thermo\", \"halo\"]:\n",
     "    name = \"_\".join(([prefix] if prefix else []) + [\"ssl\"])\n",
-    "    da = compute_ssl_timeseries(*request_reanalysis, prefix=prefix, **download_kwargs)\n",
+    "    print(f\"{name = }\")\n",
+    "    da = compute_ssl_timeseries(\n",
+    "        *request_reanalysis,\n",
+    "        prefix=prefix,\n",
+    "        lon_slice=lon_slice,\n",
+    "        lat_slice=lat_slice,\n",
+    "        **download_kwargs,\n",
+    "    )\n",
     "    dataarrays.append(da.rename(name))\n",
     "ds_reanalysis = xr.merge(dataarrays)\n",
     "\n",