Splitting a string by list of indices

后端 未结 3 512
说谎
说谎 2020-11-29 06:11

I want to split a string by a list of indices, where the split segments begin with one index and end before the next one.

Example:

         


        
3条回答
  •  情话喂你
    2020-11-29 06:26

    Here is a short solution with heavy usage of the itertools module. The tee function is used to iterate pairwise over the indices. See the Recipe section in the module for more help.

    >>> from itertools import tee, izip_longest
    >>> s = 'long string that I want to split up'
    >>> indices = [0,5,12,17]
    >>> start, end = tee(indices)
    >>> next(end)
    0
    >>> [s[i:j] for i,j in izip_longest(start, end)]
    ['long ', 'string ', 'that ', 'I want to split up']
    

    Edit: This is a version that does not copy the indices list, so it should be faster.

提交回复
热议问题