mirror of
https://github.com/python/cpython.git
synced 2025-08-03 00:23:06 +00:00
bpo-31270: Modification of Pr 3200 (#3427)
* bpo-31270: Simplify documentation of itertools.zip_longest * Use repeat(). Track num_active.
This commit is contained in:
parent
0c72a0c449
commit
3147b0422c
1 changed files with 18 additions and 19 deletions
|
@ -630,26 +630,25 @@ loops that truncate the stream.
|
|||
iterables are of uneven length, missing values are filled-in with *fillvalue*.
|
||||
Iteration continues until the longest iterable is exhausted. Roughly equivalent to::
|
||||
|
||||
class ZipExhausted(Exception):
|
||||
pass
|
||||
|
||||
def zip_longest(*args, **kwds):
|
||||
def zip_longest(*args, fillvalue=None):
|
||||
# zip_longest('ABCD', 'xy', fillvalue='-') --> Ax By C- D-
|
||||
fillvalue = kwds.get('fillvalue')
|
||||
counter = len(args) - 1
|
||||
def sentinel():
|
||||
nonlocal counter
|
||||
if not counter:
|
||||
raise ZipExhausted
|
||||
counter -= 1
|
||||
yield fillvalue
|
||||
fillers = repeat(fillvalue)
|
||||
iterators = [chain(it, sentinel(), fillers) for it in args]
|
||||
try:
|
||||
while iterators:
|
||||
yield tuple(map(next, iterators))
|
||||
except ZipExhausted:
|
||||
pass
|
||||
iterators = [iter(it) for it in args]
|
||||
num_active = len(iterators)
|
||||
if not num_active:
|
||||
return
|
||||
while True:
|
||||
values = []
|
||||
for i, it in enumerate(iterators):
|
||||
try:
|
||||
value = next(it)
|
||||
except StopIteration:
|
||||
num_active -= 1
|
||||
if not num_active:
|
||||
return
|
||||
iterators[i] = repeat(fillvalue)
|
||||
value = fillvalue
|
||||
values.append(value)
|
||||
yield tuple(values)
|
||||
|
||||
If one of the iterables is potentially infinite, then the :func:`zip_longest`
|
||||
function should be wrapped with something that limits the number of calls
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue