]> granicus.if.org Git - python/commitdiff
Add comment and improve variable name in roundrobin() (#4486)
authorRaymond Hettinger <rhettinger@users.noreply.github.com>
Tue, 21 Nov 2017 08:23:34 +0000 (00:23 -0800)
committerGitHub <noreply@github.com>
Tue, 21 Nov 2017 08:23:34 +0000 (00:23 -0800)
Doc/library/itertools.rst

index fa6c340bb7bb3edb786ecd51f911d1c997815325..a91894059d00075b5c9832a8fd81e3843acdd50a 100644 (file)
@@ -753,15 +753,16 @@ which incur interpreter overhead.
    def roundrobin(*iterables):
        "roundrobin('ABC', 'D', 'EF') --> A D E B F C"
        # Recipe credited to George Sakkis
-       pending = len(iterables)
+       num_active = len(iterables)
        nexts = cycle(iter(it).__next__ for it in iterables)
-       while pending:
+       while num_active:
            try:
                for next in nexts:
                    yield next()
            except StopIteration:
-               pending -= 1
-               nexts = cycle(islice(nexts, pending))
+               # Remove the iterator we just exhausted from the cycle.
+               num_active -= 1
+               nexts = cycle(islice(nexts, num_active))
 
    def partition(pred, iterable):
        'Use a predicate to partition entries into false entries and true entries'