Future-proof helper function with zero handling.

The ZeroDivisionError was already being caught in correlation(), but it is nicer to handle the zero case inside the helper function.
rhettinger · Aug 9, 2023 · c27f174 · c27f174
1 parent 925bbc2
commit c27f174
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Lib/statistics.py b/Lib/statistics.py
@@ -1009,6 +1009,8 @@ def _sqrtprod(x: float, y: float) -> float:
  # Square root differential correction:
  # https://www.wolframalpha.com/input/?i=Maclaurin+series+sqrt%28h**2+%2B+x%29+at+x%3D0
  h = sqrt(x * y)
+ if not h:
+ return 0.0
  x = sumprod((x, h), (y, -h))
  return h + x / (2.0 * h)