Small speed-up for convolve() recipe.

rhettinger · Jul 3, 2023 · 4c4bc60 · 4c4bc60
1 parent 2028a4f
commit 4c4bc60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Doc/library/itertools.rst b/Doc/library/itertools.rst
@@ -1085,8 +1085,8 @@ The following recipes have a more mathematical flavor:
  kernel = tuple(kernel)[::-1]
  n = len(kernel)
  padded_signal = chain(repeat(0, n-1), signal, repeat(0, n-1))
- for window in sliding_window(padded_signal, n):
-  yield math.sumprod(kernel, window)
+ windowed_signal = sliding_window(padded_signal, n)
+ return map(math.sumprod, repeat(kernel), windowed_signal)
 
  def polynomial_from_roots(roots):
  """Compute a polynomial's coefficients from its roots.