Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-11-04 07:31:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

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

Browse source
This commit is contained in:
Raymond Hettinger 2024-05-31 17:08:55 -05:00 • committed by GitHub
parent d28afd3fa0
commit cc5cd4d93e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
Lib/statistics.py
View file

@ -505,13 +505,11 @@ def fmean(data, weights=None):
n = len(data) n = len(data)
except TypeError: except TypeError:
# Handle iterators that do not define __len__(). # Handle iterators that do not define __len__().
n = 0 counter = count()
def count(iterable): total = fsum(map(itemgetter(0), zip(data, counter)))
nonlocal n n = next(counter)
for n, x in enumerate(iterable, start=1): else:
yield x total = fsum(data)
data = count(data)
total = fsum(data)
if not n: if not n:
raise StatisticsError('fmean requires at least one data point') raise StatisticsError('fmean requires at least one data point')
return total / n return total / n