import numpy as np

print(np.eye(4))