#Change the value 3 to the size of the identity matrix >>>np.identity(3) array([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])