Splitting Array In Python Interview