create a identity matrix in python code example

Example: numpy identity matrix

>>> np.identity(3)
array([[1.,  0.,  0.],
       [0.,  1.,  0.],
       [0.,  0.,  1.]])