Skip to content

Minor improvements to the docs for itertools.tee() #119135

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
rhettinger merged 4 commits into from
May 18, 2024
Merged

Minor improvements to the docs for itertools.tee() #119135

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
6f7d1e9
Move try/except out of the inner loop.
rhettinger May 17, 2024
3bba6d8
Rename to "shared_link"
rhettinger May 18, 2024
0758bbe
Simpler but slightly slower code
rhettinger May 18, 2024
8a8a142
.
rhettinger May 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
Move try/except out of the inner loop.
  • Loading branch information
@rhettinger
rhettinger committed May 17, 2024
commit 6f7d1e9425251563c4c68578c04fa0c7a83af3bc
18 changes: 9 additions & 9 deletions Doc/library/itertools.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -698,18 +698,18 @@ loops that truncate the stream.

def tee(iterable, n=2):
iterator = iter(iterable)
empty_link = [None, None] # Singly linked list: [value, link]
return tuple(_tee(iterator, empty_link) for _ in range(n))
initial_link = [None, None] # node: [data, link]
return tuple(_tee(iterator, initial_link) for _ in range(n))

def _tee(iterator, link):
while True:
if link[1] is None:
try:
try:
while True:
if link[1] is None:
link[:] = [next(iterator), [None, None]]
except StopIteration:
return
value, link = link
yield value
value, link = link
yield value
except StopIteration:
return

Once a :func:`tee` has been created, the original *iterable* should not be
used anywhere else; otherwise, the *iterable* could get advanced without
Expand Down