From b4c39035eb294a0934739764db063a2993898e2e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kai=20M=C3=BChlbauer?= Date: Wed, 6 Nov 2024 09:01:14 +0100 Subject: [PATCH] FIX: add packages, fix recipe4.ipynb --- ci/requirements/notebooks.yml | 2 ++ notebooks/workflow/recipe4.ipynb | 48 +++++++++++--------------------- 2 files changed, 19 insertions(+), 31 deletions(-) diff --git a/ci/requirements/notebooks.yml b/ci/requirements/notebooks.yml index 3fe964a5..6d63de8b 100644 --- a/ci/requirements/notebooks.yml +++ b/ci/requirements/notebooks.yml @@ -15,6 +15,8 @@ dependencies: - h5py - h5netcdf - lat_lon_parser + - libgdal-hdf5 + - libgdal-netcdf - nbconvert - nc-time-axis - netCDF4 diff --git a/notebooks/workflow/recipe4.ipynb b/notebooks/workflow/recipe4.ipynb index 0c21d714..a6672417 100644 --- a/notebooks/workflow/recipe4.ipynb +++ b/notebooks/workflow/recipe4.ipynb @@ -243,7 +243,6 @@ " tolerance=1.0, start_angle=0, stop_angle=360, angle_res=1.0, direction=1\n", ")\n", "for i, (r, v) in enumerate(zip(volume_reflectivity, volume_velocity)):\n", - " # print(i)\n", " ds0 = [\n", " xr.open_dataset(\n", " r0,\n", @@ -254,6 +253,7 @@ " )\n", " for r0 in r\n", " ]\n", + " ds0 = [r0.assign_coords(sweep_mode=r0.sweep_mode.min()) for r0 in ds0]\n", " ds1 = [\n", " xr.open_dataset(\n", " v0,\n", @@ -264,19 +264,15 @@ " )\n", " for v0 in v\n", " ]\n", - " ds = xr.concat(\n", - " [\n", - " xr.merge([r0, v0], compat=\"override\")\n", - " .assign(volume_time=r0.time.min().dt.floor(\"5min\"))\n", - " .assign_coords(sweep_mode=ds.sweep_mode.min())\n", - " .wrl.georef.georeference()\n", - " for r0, v0 in zip(ds0, ds1)\n", - " ],\n", - " \"volume_time\",\n", - " )\n", - " # ds = ds.assign_coords(sweep_mode=ds.sweep_mode.min())\n", - " # display(ds)\n", - " # ds = ds.wrl.georef.georeference()\n", + " ds1 = [r1.assign_coords(sweep_mode=r1.sweep_mode.min()) for r1 in ds1]\n", + " ds2 = [\n", + " xr.merge([r0, v0], compat=\"no_conflicts\").assign(\n", + " volume_time=r0.time.min().dt.floor(\"5min\")\n", + " )\n", + " for r0, v0 in zip(ds0, ds1)\n", + " ]\n", + " ds2 = [r2.wrl.georef.georeference() for r2 in ds2]\n", + " ds = xr.concat(ds2, \"volume_time\")\n", " dsl.append(ds)" ] }, @@ -286,13 +282,7 @@ "metadata": {}, "outputs": [], "source": [ - "# dsl.insert(0, xr.open_dataset(volume_reflectivity[0][0], group=\"/\"))\n", - "dtree = {\"/\": xd.io.backends.common._get_required_root_dataset(dsl, optional=False)}\n", - "for i, swp in enumerate(dsl):\n", - " dsl[i][\"sweep_number\"] = i\n", - " # dsl[i][\"volume_time\"] = dsl[0].volume_time\n", - "dtree = xd.io.backends.common._attach_sweep_groups(dtree, dsl)\n", - "# dtree[\"/\"][\"volume_time\"] = dsl[0].volume_time" + "dsl[0]" ] }, { @@ -300,7 +290,12 @@ "execution_count": null, "metadata": {}, "outputs": [], - "source": [] + "source": [ + "dtree = {\"/\": xd.io.backends.common._get_required_root_dataset(dsl, optional=False)}\n", + "for i, swp in enumerate(dsl):\n", + " dsl[i][\"sweep_number\"] = i\n", + "dtree = xd.io.backends.common._attach_sweep_groups(dtree, dsl)" + ] }, { "cell_type": "code", @@ -453,15 +448,6 @@ "swp = vol0[\"sweep_9\"].ds" ] }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ - "swp" - ] - }, { "cell_type": "code", "execution_count": null,