Convert 1d Array To 2d Python