import numpy as np print(np.eye(3))