Added function to get identity matrix

Author: VarunS2002

matrix_operations.py

@@ -26,6 +26,14 @@ def __get_dimensions(matrix_1: matrix, matrix_2: Optional[matrix] = None) -> \
         else:
             return dim_1
 
+    # noinspection PyUnusedLocal
+    @staticmethod
+    def identity(dimension: int) -> matrix:
+        identity_m: matrix = [[0 for cols in range(dimension)] for rows in range(dimension)]
+        for i in range(0, dimension):
+            identity_m[i][i] = 1
+        return identity_m
+
     # noinspection PyUnusedLocal
     @staticmethod
     def addition(matrix_1: matrix, matrix_2: matrix) -> matrix:
@@ -117,3 +125,5 @@ def display(matrix_1: matrix) -> None:
     MatrixOperations.display(difference)
     print('Multiplication of Matrix 3 & 4:')
     MatrixOperations.display(product)
+    print('Identity Matrix:')
+    MatrixOperations.display(MatrixOperations.identity(3))