statistics.fmean(): speed-up code path for non-sizeable inputs. (gh-119876)

rhettinger · web-flow · commit cc5cd4d93e3e · 2024-05-31T17:08:55.000-05:00
diff --git a/Lib/statistics.py b/Lib/statistics.py
@@ -505,13 +505,11 @@ def fmean(data, weights=None):
             n = len(data)
         except TypeError:
             # Handle iterators that do not define __len__().
-            n = 0
-            def count(iterable):
-                nonlocal n
-                for n, x in enumerate(iterable, start=1):
-                    yield x
-            data = count(data)
-        total = fsum(data)
+            counter = count()
+            total = fsum(map(itemgetter(0), zip(data, counter)))
+            n = next(counter)
+        else:
+            total = fsum(data)
         if not n:
             raise StatisticsError('fmean requires at least one data point')
         return total / n
