From 057b47fe4f0db119213ef3917a70ebeda5038f41 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Thu, 12 Dec 2019 14:40:08 +0100
Subject: [PATCH] get transformation information (mean, std, method) from data
prep class
---
src/data_handling/data_preparation.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/src/data_handling/data_preparation.py b/src/data_handling/data_preparation.py
index badd75aa..db800f5e 100644
--- a/src/data_handling/data_preparation.py
+++ b/src/data_handling/data_preparation.py
@@ -249,6 +249,17 @@ class DataPrep(object):
else:
self.inverse_transform()
+ def get_transformation_information(self, variable):
+ try:
+ mean = self.mean.sel({'variables': variable}).values
+ except AttributeError:
+ mean = None
+ try:
+ std = self.std.sel({'variables': variable}).values
+ except AttributeError:
+ std = None
+ return mean, std, self._transform_method
+
def make_history_window(self, dim: str, window: int) -> None:
"""
This function uses shifts the data window+1 times and returns a xarray which has a new dimension 'window'
--
GitLab