Added native Slicing and Indexing support

Abhi-1U · Dec 22, 2020 · a6ea957 · a6ea957
1 parent 2323ff9
commit a6ea957
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/pyrix/matrix/Matrix.py b/pyrix/matrix/Matrix.py
@@ -205,6 +205,22 @@ def __getattr__(self, name):
 
     def __delattr__(self, name):
         del self.__dict__[name]
+
+    def __getitem__(self,key):
+        if(isinstance(key, int)):
+            return self.matrix.data[key]
+        if(isinstance(key,(list,tuple)) and len(key)==2):
+            return self.matrix.data[key[0]][key[1]]
+        else:
+            print("MultiDimensional Matrix in Works")
+
+    def __setitem__(self,key,value):
+        if(isinstance(key, int)):
+            self.matrix.data[key]=value
+        if(isinstance(key,(list,tuple)) and len(key)==2):
+            self.matrix.data[key[0]][key[1]]=value
+        else:
+            print("MultiDimensional Matrix in Works")
     # *------- Basic Operations on Matrices -----------------------------------*
 
     # *------- Add Matrix -----------------------------------------------------*