Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2026-04-26 13:50:54 +00:00
Code Issues Projects Releases Packages Wiki Activity

Speed up counting in statistics.fmean() (gh-148875)

Browse source
This commit is contained in:
Raymond Hettinger 2026-04-22 22:06:56 -05:00 • committed by GitHub
parent 8bf99ae3a9
commit fbc7676df6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Lib/statistics.py
View file

@ -136,7 +136,7 @@
from fractions import Fraction
from decimal import Decimal
from itertools import count, groupby, repeat
from itertools import compress, count, groupby, repeat
from bisect import bisect_left, bisect_right
from math import hypot, sqrt, fabs, exp, erfc, tau, log, fsum, sumprod
from math import isfinite, isinf, pi, cos, sin, tan, cosh, asin, atan, acos
@ -195,9 +195,9 @@ def fmean(data, weights=None):
n = len(data)
except TypeError:
# Handle iterators that do not define __len__().
counter = count()
total = fsum(map(itemgetter(0), zip(data, counter)))
n = next(counter)
counter = count(1)
total = fsum(compress(data, counter))
n = next(counter) - 1
else:
total = fsum(data)