diff options
Diffstat (limited to 'szilagyi/_dataset/__init__.py')
-rw-r--r-- | szilagyi/_dataset/__init__.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/szilagyi/_dataset/__init__.py b/szilagyi/_dataset/__init__.py index cfa1aee..c91abc3 100644 --- a/szilagyi/_dataset/__init__.py +++ b/szilagyi/_dataset/__init__.py @@ -2,12 +2,20 @@ import csv import os import re -from ..nomogram import Vector - ROOT = os.path.dirname(os.path.abspath(__file__)) +class Vector(complex): + def __getitem__(self, index): + if index == 0: + return self.real + elif index == 1: + return self.imag + else: + raise IndexError + + def load(): def _read(iterable): for x, y in iterable: |