Transposing An Array In Python Interview