import numpy as np
a1 = np.array([i for i in range(1,5)])
a2 = a1.reshape((2,2))
print(f"a1 = {a1} "
f"a2 = {a2} "
)