Split Numpy Array Into Columns In Architecture