Creating A 3d Array In Numpy