Vector now supports x and y properties to wram real and imag

1 files changed, 15 insertions, 6 deletions

diff --git a/szilagyi/_dataset/__init__.py b/szilagyi/_dataset/__init__.py
index e91251e..18eaee8 100644
--- a/szilagyi/_dataset/__init__.py
+++ b/szilagyi/_dataset/__init__.py
@@ -4,12 +4,21 @@ import re
 
 
 class Vector(complex):
-	def __getitem__(self, index):
-		if index == 0:
-			return self.real
-		if index == 1:
-			return self.imag
-		raise IndexError
+	@property
+	def x(self):
+		return self.real
+
+	@x.setter
+	def x(self, value):
+		self.real = value
+
+	@property
+	def y(self):
+		return self.imag
+
+	@y.setter
+	def y(self, value):
+		self.imag = value
 
 
 def _load_from(root):