From 3dd15c90a814320bc26cbaec0f75e346b9eb4db3 Mon Sep 17 00:00:00 2001
From: "v.gramlich1" <v.gramlichfz-juelich.de>
Date: Tue, 24 Aug 2021 20:18:03 +0200
Subject: [PATCH] Fix in the default_data_handler.py in multiply_extremes
---
mlair/data_handler/default_data_handler.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/mlair/data_handler/default_data_handler.py b/mlair/data_handler/default_data_handler.py
index c97d57ef..2ffdd49e 100644
--- a/mlair/data_handler/default_data_handler.py
+++ b/mlair/data_handler/default_data_handler.py
@@ -273,14 +273,16 @@ class DefaultDataHandler(AbstractDataHandler):
else:
extreme_idx = xr.concat([(Y < -extr_val).any(dim=other_dims[0]),
(Y > extr_val).any(dim=other_dims[0])],
- dim=other_dims[1]).any(dim=other_dims[1])
+ dim=other_dims[0]).any(dim=other_dims[0])
- extremes_X = list(map(lambda x: x.sel(**{dim: extreme_idx}), X))
+ sel = extreme_idx[extreme_idx].coords[dim].values
+ extremes_X = list(map(lambda x: x.sel(**{dim: sel}), X))
self._add_timedelta(extremes_X, dim, timedelta)
# extremes_X = list(map(lambda x: x.coords[dim].values + np.timedelta64(*timedelta), extremes_X))
extremes_Y = Y.sel(**{dim: extreme_idx})
- extremes_Y.coords[dim].values += np.timedelta64(*timedelta)
+ #extremes_Y.coords[dim].values += np.timedelta64(*timedelta)
+ self._add_timedelta(extremes_Y, dim, timedelta)
self._Y_extreme = xr.concat([Y, extremes_Y], dim=dim)
self._X_extreme = list(map(lambda x1, x2: xr.concat([x1, x2], dim=dim), X, extremes_X))
--
GitLab