import numpy as np

print(np.eye(3))
print(np.eye(4))
print(np.eye(5))