From 931280e4690f96be4000e23cc78c99bd8b1285cf Mon Sep 17 00:00:00 2001
From: madsmpedersen <m@madsp.dk>
Date: Fri, 22 Jul 2016 08:03:24 +0200
Subject: [PATCH] added Dataset class

---
 wetb/gtsdf/__init__.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/wetb/gtsdf/__init__.py b/wetb/gtsdf/__init__.py
index 0ba32914..b4b88948 100644
--- a/wetb/gtsdf/__init__.py
+++ b/wetb/gtsdf/__init__.py
@@ -39,6 +39,15 @@ from .gtsdf import load
 from .gtsdf import append_block
 from .gtsdf import load_pandas
 
+class Dataset(object):
+    def __init__(self, filename):
+        self.time, self.data, self.info = load(filename)
+    def __call__(self, name):
+        if name == "Time":
+            return self.time
+        return self.data[:, self.info['attribute_names'].index(name)]
+
+
 __all__ = sorted([m for m in set(dir()) - set(d)])
 
 
-- 
GitLab