import numpy as np

print(np.eye(3))

#kochm programować