]> granicus.if.org Git - python/commitdiff
bpo-32099 Add deque variant of roundrobin() recipe (#4497)
authorRaymond Hettinger <rhettinger@users.noreply.github.com>
Thu, 23 Nov 2017 21:32:23 +0000 (13:32 -0800)
committerGitHub <noreply@github.com>
Thu, 23 Nov 2017 21:32:23 +0000 (13:32 -0800)
* Minor wording tweaks

Doc/library/collections.rst

index cda829694a32774500f8f5ab1c78fd9a27f7ca22..4b0d8c048ae7d9aba04477a3ec44a7ab8094a1d1 100644 (file)
@@ -618,6 +618,25 @@ added elements by appending to the right and popping to the left::
             d.append(elem)
             yield s / n
 
+A `round-robin scheduler
+<https://en.wikipedia.org/wiki/Round-robin_scheduling>`_ can be implemented with
+input iterators stored in a :class:`deque`.  Values are yielded from the active
+iterator in position zero.  If that iterator is exhausted, it can be removed
+with :meth:`~deque.popleft`; otherwise, it can be cycled back to the end with
+the :meth:`~deque.rotate` method::
+
+    def roundrobin(*iterables):
+        "roundrobin('ABC', 'D', 'EF') --> A D E B F C"
+        iterators = deque(map(iter, iterables))
+        while iterators:
+            try:
+                while True:
+                    yield next(iterators[0])
+                    iterators.rotate(-1)
+            except StopIteration:
+                # Remove an exhausted iterator.
+                iterators.popleft()
+
 The :meth:`rotate` method provides a way to implement :class:`deque` slicing and
 deletion.  For example, a pure Python implementation of ``del d[n]`` relies on
 the :meth:`rotate` method to position elements to be popped::