np identity like code example Example: numpy identity matrix >>> np.identity(3) array([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])