bpo-24567: Random subnormal.diff (#7954)

Handle subnormal weights for choices()
This commit is contained in:
Raymond Hettinger 2018-06-27 01:08:31 -07:00 committed by GitHub
parent 3c8043d8fa
commit ddf7171911
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 13 additions and 1 deletions

View file

@ -383,7 +383,9 @@ class Random(_random.Random):
raise ValueError('The number of weights does not match the population')
bisect = _bisect.bisect
total = cum_weights[-1]
return [population[bisect(cum_weights, random() * total)] for i in range(k)]
hi = len(cum_weights) - 1
return [population[bisect(cum_weights, random() * total, 0, hi)]
for i in range(k)]
## -------------------- real-valued distributions -------------------