diff options
author | Raymond Hettinger <rhettinger@users.noreply.github.com> | 2019-02-16 21:30:51 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-02-16 21:30:51 (GMT) |
commit | 5382203ae1bf637f9554277d5597d04805934420 (patch) | |
tree | ebfd86efe1b1edac1e4fab0710d638f4144c47db /Lib/random.py | |
parent | 38f11cc3f62db11a4a24354bd06273322ac91afa (diff) | |
download | cpython-5382203ae1bf637f9554277d5597d04805934420.zip cpython-5382203ae1bf637f9554277d5597d04805934420.tar.gz cpython-5382203ae1bf637f9554277d5597d04805934420.tar.bz2 |
Convert range to repeat for choices() (#11889)
Diffstat (limited to 'Lib/random.py')
-rw-r--r-- | Lib/random.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/Lib/random.py b/Lib/random.py index 754b295..79ef30d 100644 --- a/Lib/random.py +++ b/Lib/random.py @@ -43,7 +43,7 @@ from math import sqrt as _sqrt, acos as _acos, cos as _cos, sin as _sin from os import urandom as _urandom from _collections_abc import Set as _Set, Sequence as _Sequence from hashlib import sha512 as _sha512 -from itertools import accumulate as _accumulate +from itertools import accumulate as _accumulate, repeat as _repeat from bisect import bisect as _bisect import os as _os @@ -389,7 +389,7 @@ class Random(_random.Random): if weights is None: _int = int n += 0.0 # convert to float for a small speed improvement - return [population[_int(random() * n)] for i in range(k)] + return [population[_int(random() * n)] for i in _repeat(None, k)] cum_weights = list(_accumulate(weights)) elif weights is not None: raise TypeError('Cannot specify both weights and cumulative weights') @@ -399,7 +399,7 @@ class Random(_random.Random): total = cum_weights[-1] + 0.0 # convert to float hi = n - 1 return [population[bisect(cum_weights, random() * total, 0, hi)] - for i in range(k)] + for i in _repeat(None, k)] ## -------------------- real-valued distributions ------------------- |