From 7477d0bc9a8622b0e51685fa355605382d3e9187 Mon Sep 17 00:00:00 2001 From: Dennis Noll <dennis.noll@rwth-aachen.de> Date: Fri, 28 Aug 2020 07:46:54 +0200 Subject: [PATCH] [data] DSS: adds concantenate --- data.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/data.py b/data.py index 6044750..24f8222 100644 --- a/data.py +++ b/data.py @@ -67,6 +67,10 @@ class SKDict(dict): keys.update(*(inst.keys() for inst in insts)) return cls({key: tuple(inst.get(key) for inst in insts) for key in keys}) + @classmethod + def concatenate(cls, *insts): + return cls.zip(*insts).map(lambda x: np.concatenate(x)) + def only(self, *keys): return self.__class__({key: self[key] for key in keys}) -- GitLab