summaryrefslogtreecommitdiffstats
path: root/Lib/random.py
diff options
context:
space:
mode:
authorRaymond Hettinger <rhettinger@users.noreply.github.com>2019-02-16 21:30:51 (GMT)
committerGitHub <noreply@github.com>2019-02-16 21:30:51 (GMT)
commit5382203ae1bf637f9554277d5597d04805934420 (patch)
treeebfd86efe1b1edac1e4fab0710d638f4144c47db /Lib/random.py
parent38f11cc3f62db11a4a24354bd06273322ac91afa (diff)
downloadcpython-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.py6
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 -------------------