mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-09-30 12:21:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

[3.12] Improve all_equal() recipe (gh-116081) (gh-116083)

Browse source
This commit is contained in:
Miss Islington (bot) 2024-02-29 00:11:46 +01:00 • committed by GitHub
parent dec637a953
commit a85e512699
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
Doc/library/itertools.rst
View file

@ -855,10 +855,9 @@ which incur interpreter overhead.
"Given a predicate that returns True or False, count the True results."
return sum(map(pred, iterable))
def all_equal(iterable):
def all_equal(iterable, key=None):
"Returns True if all the elements are equal to each other."
g = groupby(iterable)
return next(g, True) and not next(g, False)
return len(take(2, groupby(iterable, key))) <= 1
def first_true(iterable, default=False, pred=None):
"""Returns the first true value in the iterable.
@ -1217,6 +1216,8 @@ The following recipes have a more mathematical flavor:
>>> [all_equal(s) for s in ('', 'A', 'AAAA', 'AAAB', 'AAABA')]
[True, True, True, False, False]
>>> [all_equal(s, key=str.casefold) for s in ('', 'A', 'AaAa', 'AAAB', 'AAABA')]
[True, True, True, False, False]
>>> quantify(range(99), lambda x: x%2==0)
50