diff --git a/wetb/gtsdf/__init__.py b/wetb/gtsdf/__init__.py index 0ba329148e5bf2a39f22b1e821d4315e6f1172f8..b4b88948e4f83b7cde6d4d35ee419a8d660aa64b 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)])