Reshape 2d Array To 3d Python