Cycle around a sequence

Terry Reedy tjreedy at udel.edu
Wed Feb 8 15:15:59 EST 2012


On 2/8/2012 9:25 AM, Neil Cerutti wrote:
> On 2012-02-08, Mark Lawrence<breamoreboy at yahoo.co.uk>  wrote:
>> I'm looking at a way of cycling around a sequence i.e. starting
>> at some given location in the middle of a sequence and running
>> to the end before coming back to the beginning and running to
>> the start place.  About the best I could come up with is the
>> following, any better ideas for some definition of better?
>
> Python's indices were designed for these kinds of shenanigans.
>
> def rotated(seq, n):
>      """Iterate through all of seq, but starting from index n.
>
>      >>>  ", ".join(str(n) for n in rotated(range(5), 3))
>      '3, 4, 0, 1, 2'
>      """
>
>      i = n - len(seq)
>      while i<  n:
>          yield seq[i]
>          i += 1

This is really nice, in the category of "Why didn't I think of that?"
(Probably because I knew the % mod solution from C and never 'updated'!)

> if __name__ == "__main__":
>      import doctest
>      doctest.testmod()
>
> If you have merely an iterable instead of a sequence, then look
> to some of the other clever stuff already posted.

To make a repeating rotator is only a slight adjustment:

     k = n - len(seq)
     while True:
         i = k
         while i < n:
             yield seq[i]
             i += 1

-- 
Terry Jan Reedy




More information about the Python-list mailing list