import numpy as np print(np.eye(4)) print('///////////') print(np.eye(10))