From 36ea21d6a424ba325cdb9d8608989e58463f9090 Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Sat, 15 Oct 2022 15:28:56 -0500 Subject: [PATCH 1/2] Improve speed. Reduce auxiliary memory load to 16.6% of the main array. --- Doc/library/itertools.rst | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/Doc/library/itertools.rst b/Doc/library/itertools.rst index 056b0788a4d846..36930a7567c0d5 100644 --- a/Doc/library/itertools.rst +++ b/Doc/library/itertools.rst @@ -822,12 +822,13 @@ which incur interpreter overhead. def sieve(n): "Primes less than n" # sieve(30) --> 2 3 5 7 11 13 17 19 23 29 - data = bytearray([1]) * n - data[:2] = 0, 0 + data = bytearray((0, 1)) * (n // 2) + data[:3] = 0, 0, 0 limit = math.isqrt(n) + 1 for p in compress(range(limit), data): - data[p*p : n : p] = bytearray(len(range(p*p, n, p))) - return iter_index(data, 1) + data[p*p : n : p+p] = bytearray(len(range(p*p, n, p+p))) + data[2] = 1 + return iter_index(data, 1) if n > 2 else iter([]) def flatten(list_of_lists): "Flatten one level of nesting" From ec05c8a4e79460308a643ebe206d8556044c6895 Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Sat, 15 Oct 2022 15:48:47 -0500 Subject: [PATCH 2/2] Zeros array is immutable --- Doc/library/itertools.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Doc/library/itertools.rst b/Doc/library/itertools.rst index 36930a7567c0d5..88e1e5aa6ef7f3 100644 --- a/Doc/library/itertools.rst +++ b/Doc/library/itertools.rst @@ -826,7 +826,7 @@ which incur interpreter overhead. data[:3] = 0, 0, 0 limit = math.isqrt(n) + 1 for p in compress(range(limit), data): - data[p*p : n : p+p] = bytearray(len(range(p*p, n, p+p))) + data[p*p : n : p+p] = bytes(len(range(p*p, n, p+p))) data[2] = 1 return iter_index(data, 1) if n > 2 else iter([])