Python intro notes

2019/PythonIntro/notes/04_2_makingObjects.py

@@ -0,0 +1,32 @@
+"""
+Skippable if out of time, hot to define ones own objects.
+"""
+
+from __future__ import annotations
+
+class MyVector(list):
+    a = 0
+
+    def __init__(self, dim: int, data: list):
+        if dim != len(data):
+            raise ValueError("Wrong dimensions in the vector!!!")
+        self.dim = dim
+        self.data = data
+    
+    def __add__(self, other: MyVector):
+        if self.dim != other.dim:
+            return None
+        else:
+            return MyVector(self.dim, [a + b for a, b in zip(self.data, other.data)])
+    
+    def __repr__(self):
+        return "A vector of dimension {} with data:\n{}".format(self.dim, self.data)
+
+
+a = MyVector(3, [1, 2, 3])
+b = MyVector(3, [-1, -4, 8])
+
+print(a)
+print(b)
+
+print(a + b)