Use match/case in grouper() recipe (gh-113059)

Use match/case in grouper() reciper
This commit is contained in:
Raymond Hettinger 2023-12-13 12:11:52 -06:00 committed by GitHub
parent 480b4b359d
commit 2111795d0c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -914,14 +914,15 @@ which incur interpreter overhead.
# grouper('ABCDEFG', 3, incomplete='strict') --> ABC DEF ValueError
# grouper('ABCDEFG', 3, incomplete='ignore') --> ABC DEF
args = [iter(iterable)] * n
if incomplete == 'fill':
return zip_longest(*args, fillvalue=fillvalue)
elif incomplete == 'strict':
return zip(*args, strict=True)
elif incomplete == 'ignore':
return zip(*args)
else:
raise ValueError('Expected fill, strict, or ignore')
match incomplete:
case 'fill':
return zip_longest(*args, fillvalue=fillvalue)
case 'strict':
return zip(*args, strict=True)
case 'ignore':
return zip(*args)
case _:
raise ValueError('Expected fill, strict, or ignore')
def sliding_window(iterable, n):
# sliding_window('ABCDEFG', 4) --> ABCD BCDE CDEF DEFG