ecmwf
diff --git a/‎packages/evaluate/src/weathergen/evaluate/io_reader.py‎
Lines changed: 7 additions & 1 deletion b/‎packages/evaluate/src/weathergen/evaluate/io_reader.py‎
Lines changed: 7 additions & 1 deletion
diff --git a/‎packages/evaluate/src/weathergen/evaluate/plotter.py‎
Lines changed: 71 additions & 2 deletions b/‎packages/evaluate/src/weathergen/evaluate/plotter.py‎
Lines changed: 71 additions & 2 deletions
@@ -164,7 +164,6 @@ def check_availability(
         fsteps = requested_data.fsteps
         samples = requested_data.samples
         ensemble = requested_data.ensemble
-
         requested = {
             "channel": set(channels) if channels is not None else None,
             "fstep": set(fsteps) if fsteps is not None else None,
@@ -478,6 +477,13 @@ def get_data(
                         _logger.debug(f"Selecting ensemble members {ensemble}.")
                         pred = pred.sel(ens=ensemble)
 
+                    if ensemble == ["mean"]:
+                        _logger.debug("Averaging over ensemble members.")
+                        pred = pred.mean("ens", keepdims=True)
+                    else:
+                        _logger.debug(f"Selecting ensemble members {ensemble}.")
+                        pred = pred.sel(ens=ensemble)
+
                     da_tars_fs.append(target.squeeze())
                     da_preds_fs.append(pred.squeeze())
                     pps.append(npoints)
 
@@ -138,7 +138,6 @@ def select_from_da(self, da: xr.DataArray, selection: dict) -> xr.DataArray:
         -------
             xarray DataArray with selected data.
         """
-
         for key, value in selection.items():
             if key in da.coords and key not in da.dims:
                 # Coordinate like 'sample' aligned to another dim
@@ -710,6 +709,77 @@ def _plot_ensemble(self, data: xr.DataArray, x_dim: str, label: str) -> None:
                 f"LinePlot:: Unknown option for plot_ensemble: {self.plot_ensemble}. Skipping ensemble plotting."
             )
 
+    def _plot_ensemble(self, data: xr.DataArray, x_dim: str, label: str) -> None:
+        """
+        Plot ensemble spread for a data array.
+
+        Parameters
+        ----------
+        data: xr.xArray
+            DataArray to be plotted
+        x_dim: str
+            Dimension to be used for the x-axis.
+        label: str
+            Label for the dataset
+        Returns
+        -------
+            None
+        """
+        averaged = data.mean(dim=[dim for dim in data.dims if dim != x_dim], skipna=True).sortby(
+            x_dim
+        )
+
+        lines = plt.plot(
+            averaged[x_dim],
+            averaged.values,
+            label=label,
+            marker="o",
+            linestyle="-",
+        )
+        line = lines[0]
+        color = line.get_color()
+
+        ens = data.mean(
+            dim=[dim for dim in data.dims if dim not in [x_dim, "ens"]], skipna=True
+        ).sortby(x_dim)
+
+        if self.plot_ensemble == "std":
+            std_dev = ens.std(dim="ens", skipna=True).sortby(x_dim)
+            plt.fill_between(
+                averaged[x_dim],
+                (averaged - std_dev).values,
+                (averaged + std_dev).values,
+                label=f"{label} - std dev",
+                color=color,
+                alpha=0.2,
+            )
+
+        elif self.plot_ensemble == "minmax":
+            ens_min = ens.min(dim="ens", skipna=True).sortby(x_dim)
+            ens_max = ens.max(dim="ens", skipna=True).sortby(x_dim)
+
+            plt.fill_between(
+                averaged[x_dim],
+                ens_min.values,
+                ens_max.values,
+                label=f"{label} - min max",
+                color=color,
+                alpha=0.2,
+            )
+
+        elif self.plot_ensemble == "members":
+            for j in range(ens.ens.size):
+                plt.plot(
+                    ens[x_dim],
+                    ens.isel(ens=j).values,
+                    color=color,
+                    alpha=0.2,
+                )
+        else:
+            _logger.warning(
+                f"LinePlot:: Unknown option for plot_ensemble: {self.plot_ensemble}. Skipping ensemble plotting."
+            )
+
     def plot(
         self,
         data: xr.DataArray | list,
@@ -737,7 +807,6 @@ def plot(
             Name of the dimension to be used for the y-axis.
         print_summary:
             If True, print a summary of the values from the graph.
-
         Returns
         -------
             None