Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 1 | :mod:`random` --- Generate pseudo-random numbers |
| 2 | ================================================ |
| 3 | |
| 4 | .. module:: random |
| 5 | :synopsis: Generate pseudo-random numbers with various common distributions. |
| 6 | |
Raymond Hettinger | 1048094 | 2011-01-10 03:26:08 +0000 | [diff] [blame] | 7 | **Source code:** :source:`Lib/random.py` |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 8 | |
Raymond Hettinger | 4f707fd | 2011-01-10 19:54:11 +0000 | [diff] [blame] | 9 | -------------- |
| 10 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 11 | This module implements pseudo-random number generators for various |
| 12 | distributions. |
| 13 | |
Raymond Hettinger | b21dac1 | 2010-09-07 05:32:49 +0000 | [diff] [blame] | 14 | For integers, there is uniform selection from a range. For sequences, there is |
| 15 | uniform selection of a random element, a function to generate a random |
| 16 | permutation of a list in-place, and a function for random sampling without |
| 17 | replacement. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 18 | |
| 19 | On the real line, there are functions to compute uniform, normal (Gaussian), |
| 20 | lognormal, negative exponential, gamma, and beta distributions. For generating |
| 21 | distributions of angles, the von Mises distribution is available. |
| 22 | |
| 23 | Almost all module functions depend on the basic function :func:`random`, which |
| 24 | generates a random float uniformly in the semi-open range [0.0, 1.0). Python |
| 25 | uses the Mersenne Twister as the core generator. It produces 53-bit precision |
| 26 | floats and has a period of 2\*\*19937-1. The underlying implementation in C is |
| 27 | both fast and threadsafe. The Mersenne Twister is one of the most extensively |
| 28 | tested random number generators in existence. However, being completely |
| 29 | deterministic, it is not suitable for all purposes, and is completely unsuitable |
| 30 | for cryptographic purposes. |
| 31 | |
| 32 | The functions supplied by this module are actually bound methods of a hidden |
| 33 | instance of the :class:`random.Random` class. You can instantiate your own |
Raymond Hettinger | 28de64f | 2008-01-13 23:40:30 +0000 | [diff] [blame] | 34 | instances of :class:`Random` to get generators that don't share state. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 35 | |
| 36 | Class :class:`Random` can also be subclassed if you want to use a different |
| 37 | basic generator of your own devising: in that case, override the :meth:`random`, |
Raymond Hettinger | afd3045 | 2009-02-24 10:57:02 +0000 | [diff] [blame] | 38 | :meth:`seed`, :meth:`getstate`, and :meth:`setstate` methods. |
Benjamin Peterson | d18de0e | 2008-07-31 20:21:46 +0000 | [diff] [blame] | 39 | Optionally, a new generator can supply a :meth:`getrandbits` method --- this |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 40 | allows :meth:`randrange` to produce selections over an arbitrarily large range. |
| 41 | |
Benjamin Peterson | 21896a3 | 2010-03-21 22:03:03 +0000 | [diff] [blame] | 42 | The :mod:`random` module also provides the :class:`SystemRandom` class which |
| 43 | uses the system function :func:`os.urandom` to generate random numbers |
| 44 | from sources provided by the operating system. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 45 | |
Victor Stinner | 19fb53c | 2011-05-24 21:32:40 +0200 | [diff] [blame] | 46 | .. warning:: |
| 47 | |
| 48 | The generators of the :mod:`random` module should not be used for security |
Victor Stinner | c58140c | 2011-05-25 13:13:55 +0200 | [diff] [blame] | 49 | purposes. Use :func:`ssl.RAND_bytes` if you require a cryptographically |
| 50 | secure pseudorandom number generator. |
Victor Stinner | 19fb53c | 2011-05-24 21:32:40 +0200 | [diff] [blame] | 51 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 52 | |
Raymond Hettinger | 1048094 | 2011-01-10 03:26:08 +0000 | [diff] [blame] | 53 | Bookkeeping functions: |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 54 | |
Raymond Hettinger | f763a72 | 2010-09-07 00:38:15 +0000 | [diff] [blame] | 55 | .. function:: seed([x], version=2) |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 56 | |
Raymond Hettinger | f763a72 | 2010-09-07 00:38:15 +0000 | [diff] [blame] | 57 | Initialize the random number generator. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 58 | |
Raymond Hettinger | f763a72 | 2010-09-07 00:38:15 +0000 | [diff] [blame] | 59 | If *x* is omitted or ``None``, the current system time is used. If |
| 60 | randomness sources are provided by the operating system, they are used |
| 61 | instead of the system time (see the :func:`os.urandom` function for details |
| 62 | on availability). |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 63 | |
Raymond Hettinger | f763a72 | 2010-09-07 00:38:15 +0000 | [diff] [blame] | 64 | If *x* is an int, it is used directly. |
| 65 | |
| 66 | With version 2 (the default), a :class:`str`, :class:`bytes`, or :class:`bytearray` |
Raymond Hettinger | 3149f9c | 2010-09-07 05:35:10 +0000 | [diff] [blame] | 67 | object gets converted to an :class:`int` and all of its bits are used. With version 1, |
Raymond Hettinger | f763a72 | 2010-09-07 00:38:15 +0000 | [diff] [blame] | 68 | the :func:`hash` of *x* is used instead. |
| 69 | |
| 70 | .. versionchanged:: 3.2 |
| 71 | Moved to the version 2 scheme which uses all of the bits in a string seed. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 72 | |
| 73 | .. function:: getstate() |
| 74 | |
| 75 | Return an object capturing the current internal state of the generator. This |
| 76 | object can be passed to :func:`setstate` to restore the state. |
| 77 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 78 | |
| 79 | .. function:: setstate(state) |
| 80 | |
| 81 | *state* should have been obtained from a previous call to :func:`getstate`, and |
| 82 | :func:`setstate` restores the internal state of the generator to what it was at |
| 83 | the time :func:`setstate` was called. |
| 84 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 85 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 86 | .. function:: getrandbits(k) |
| 87 | |
Ezio Melotti | 0639d5a | 2009-12-19 23:26:38 +0000 | [diff] [blame] | 88 | Returns a Python integer with *k* random bits. This method is supplied with |
Georg Brandl | 5c10664 | 2007-11-29 17:41:05 +0000 | [diff] [blame] | 89 | the MersenneTwister generator and some other generators may also provide it |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 90 | as an optional part of the API. When available, :meth:`getrandbits` enables |
| 91 | :meth:`randrange` to handle arbitrarily large ranges. |
| 92 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 93 | |
| 94 | Functions for integers: |
| 95 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 96 | .. function:: randrange([start,] stop[, step]) |
| 97 | |
| 98 | Return a randomly selected element from ``range(start, stop, step)``. This is |
| 99 | equivalent to ``choice(range(start, stop, step))``, but doesn't actually build a |
| 100 | range object. |
| 101 | |
Raymond Hettinger | 0515661 | 2010-09-07 04:44:52 +0000 | [diff] [blame] | 102 | The positional argument pattern matches that of :func:`range`. Keyword arguments |
| 103 | should not be used because the function may use them in unexpected ways. |
| 104 | |
| 105 | .. versionchanged:: 3.2 |
| 106 | :meth:`randrange` is more sophisticated about producing equally distributed |
| 107 | values. Formerly it used a style like ``int(random()*n)`` which could produce |
| 108 | slightly uneven distributions. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 109 | |
| 110 | .. function:: randint(a, b) |
| 111 | |
Raymond Hettinger | afd3045 | 2009-02-24 10:57:02 +0000 | [diff] [blame] | 112 | Return a random integer *N* such that ``a <= N <= b``. Alias for |
| 113 | ``randrange(a, b+1)``. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 114 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 115 | |
Georg Brandl | 55ac8f0 | 2007-09-01 13:51:09 +0000 | [diff] [blame] | 116 | Functions for sequences: |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 117 | |
| 118 | .. function:: choice(seq) |
| 119 | |
| 120 | Return a random element from the non-empty sequence *seq*. If *seq* is empty, |
| 121 | raises :exc:`IndexError`. |
| 122 | |
| 123 | |
| 124 | .. function:: shuffle(x[, random]) |
| 125 | |
| 126 | Shuffle the sequence *x* in place. The optional argument *random* is a |
| 127 | 0-argument function returning a random float in [0.0, 1.0); by default, this is |
| 128 | the function :func:`random`. |
| 129 | |
| 130 | Note that for even rather small ``len(x)``, the total number of permutations of |
| 131 | *x* is larger than the period of most random number generators; this implies |
| 132 | that most permutations of a long sequence can never be generated. |
| 133 | |
| 134 | |
| 135 | .. function:: sample(population, k) |
| 136 | |
Raymond Hettinger | 1acde19 | 2008-01-14 01:00:53 +0000 | [diff] [blame] | 137 | Return a *k* length list of unique elements chosen from the population sequence |
| 138 | or set. Used for random sampling without replacement. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 139 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 140 | Returns a new list containing elements from the population while leaving the |
| 141 | original population unchanged. The resulting list is in selection order so that |
| 142 | all sub-slices will also be valid random samples. This allows raffle winners |
| 143 | (the sample) to be partitioned into grand prize and second place winners (the |
| 144 | subslices). |
| 145 | |
Guido van Rossum | 2cc30da | 2007-11-02 23:46:40 +0000 | [diff] [blame] | 146 | Members of the population need not be :term:`hashable` or unique. If the population |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 147 | contains repeats, then each occurrence is a possible selection in the sample. |
| 148 | |
| 149 | To choose a sample from a range of integers, use an :func:`range` object as an |
| 150 | argument. This is especially fast and space efficient for sampling from a large |
| 151 | population: ``sample(range(10000000), 60)``. |
| 152 | |
| 153 | The following functions generate specific real-valued distributions. Function |
| 154 | parameters are named after the corresponding variables in the distribution's |
| 155 | equation, as used in common mathematical practice; most of these equations can |
| 156 | be found in any statistics text. |
| 157 | |
| 158 | |
| 159 | .. function:: random() |
| 160 | |
| 161 | Return the next random floating point number in the range [0.0, 1.0). |
| 162 | |
| 163 | |
| 164 | .. function:: uniform(a, b) |
| 165 | |
Benjamin Peterson | b58dda7 | 2009-01-18 22:27:04 +0000 | [diff] [blame] | 166 | Return a random floating point number *N* such that ``a <= N <= b`` for |
| 167 | ``a <= b`` and ``b <= N <= a`` for ``b < a``. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 168 | |
Raymond Hettinger | be40db0 | 2009-06-11 23:12:14 +0000 | [diff] [blame] | 169 | The end-point value ``b`` may or may not be included in the range |
| 170 | depending on floating-point rounding in the equation ``a + (b-a) * random()``. |
Benjamin Peterson | 35e8c46 | 2008-04-24 02:34:53 +0000 | [diff] [blame] | 171 | |
Georg Brandl | 73dd7c7 | 2011-09-17 20:36:28 +0200 | [diff] [blame] | 172 | |
Christian Heimes | fe337bf | 2008-03-23 21:54:12 +0000 | [diff] [blame] | 173 | .. function:: triangular(low, high, mode) |
| 174 | |
Benjamin Peterson | b58dda7 | 2009-01-18 22:27:04 +0000 | [diff] [blame] | 175 | Return a random floating point number *N* such that ``low <= N <= high`` and |
Christian Heimes | cc47b05 | 2008-03-25 14:56:36 +0000 | [diff] [blame] | 176 | with the specified *mode* between those bounds. The *low* and *high* bounds |
| 177 | default to zero and one. The *mode* argument defaults to the midpoint |
| 178 | between the bounds, giving a symmetric distribution. |
Christian Heimes | fe337bf | 2008-03-23 21:54:12 +0000 | [diff] [blame] | 179 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 180 | |
| 181 | .. function:: betavariate(alpha, beta) |
| 182 | |
Benjamin Peterson | b58dda7 | 2009-01-18 22:27:04 +0000 | [diff] [blame] | 183 | Beta distribution. Conditions on the parameters are ``alpha > 0`` and |
| 184 | ``beta > 0``. Returned values range between 0 and 1. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 185 | |
| 186 | |
| 187 | .. function:: expovariate(lambd) |
| 188 | |
Mark Dickinson | 2f94736 | 2009-01-07 17:54:07 +0000 | [diff] [blame] | 189 | Exponential distribution. *lambd* is 1.0 divided by the desired |
| 190 | mean. It should be nonzero. (The parameter would be called |
| 191 | "lambda", but that is a reserved word in Python.) Returned values |
| 192 | range from 0 to positive infinity if *lambd* is positive, and from |
| 193 | negative infinity to 0 if *lambd* is negative. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 194 | |
| 195 | |
| 196 | .. function:: gammavariate(alpha, beta) |
| 197 | |
Benjamin Peterson | b58dda7 | 2009-01-18 22:27:04 +0000 | [diff] [blame] | 198 | Gamma distribution. (*Not* the gamma function!) Conditions on the |
| 199 | parameters are ``alpha > 0`` and ``beta > 0``. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 200 | |
Georg Brandl | 73dd7c7 | 2011-09-17 20:36:28 +0200 | [diff] [blame] | 201 | The probability distribution function is:: |
| 202 | |
| 203 | x ** (alpha - 1) * math.exp(-x / beta) |
| 204 | pdf(x) = -------------------------------------- |
| 205 | math.gamma(alpha) * beta ** alpha |
| 206 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 207 | |
| 208 | .. function:: gauss(mu, sigma) |
| 209 | |
Benjamin Peterson | b58dda7 | 2009-01-18 22:27:04 +0000 | [diff] [blame] | 210 | Gaussian distribution. *mu* is the mean, and *sigma* is the standard |
| 211 | deviation. This is slightly faster than the :func:`normalvariate` function |
| 212 | defined below. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 213 | |
| 214 | |
| 215 | .. function:: lognormvariate(mu, sigma) |
| 216 | |
| 217 | Log normal distribution. If you take the natural logarithm of this |
| 218 | distribution, you'll get a normal distribution with mean *mu* and standard |
| 219 | deviation *sigma*. *mu* can have any value, and *sigma* must be greater than |
| 220 | zero. |
| 221 | |
| 222 | |
| 223 | .. function:: normalvariate(mu, sigma) |
| 224 | |
| 225 | Normal distribution. *mu* is the mean, and *sigma* is the standard deviation. |
| 226 | |
| 227 | |
| 228 | .. function:: vonmisesvariate(mu, kappa) |
| 229 | |
| 230 | *mu* is the mean angle, expressed in radians between 0 and 2\*\ *pi*, and *kappa* |
| 231 | is the concentration parameter, which must be greater than or equal to zero. If |
| 232 | *kappa* is equal to zero, this distribution reduces to a uniform random angle |
| 233 | over the range 0 to 2\*\ *pi*. |
| 234 | |
| 235 | |
| 236 | .. function:: paretovariate(alpha) |
| 237 | |
| 238 | Pareto distribution. *alpha* is the shape parameter. |
| 239 | |
| 240 | |
| 241 | .. function:: weibullvariate(alpha, beta) |
| 242 | |
| 243 | Weibull distribution. *alpha* is the scale parameter and *beta* is the shape |
| 244 | parameter. |
| 245 | |
| 246 | |
Raymond Hettinger | 3cdf871 | 2010-12-02 05:35:35 +0000 | [diff] [blame] | 247 | Alternative Generator: |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 248 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 249 | .. class:: SystemRandom([seed]) |
| 250 | |
| 251 | Class that uses the :func:`os.urandom` function for generating random numbers |
| 252 | from sources provided by the operating system. Not available on all systems. |
Raymond Hettinger | 3cdf871 | 2010-12-02 05:35:35 +0000 | [diff] [blame] | 253 | Does not rely on software state, and sequences are not reproducible. Accordingly, |
Raymond Hettinger | afd3045 | 2009-02-24 10:57:02 +0000 | [diff] [blame] | 254 | the :meth:`seed` method has no effect and is ignored. |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 255 | The :meth:`getstate` and :meth:`setstate` methods raise |
| 256 | :exc:`NotImplementedError` if called. |
| 257 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 258 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 259 | .. seealso:: |
| 260 | |
| 261 | M. Matsumoto and T. Nishimura, "Mersenne Twister: A 623-dimensionally |
| 262 | equidistributed uniform pseudorandom number generator", ACM Transactions on |
| 263 | Modeling and Computer Simulation Vol. 8, No. 1, January pp.3-30 1998. |
| 264 | |
Georg Brandl | 116aa62 | 2007-08-15 14:28:22 +0000 | [diff] [blame] | 265 | |
Raymond Hettinger | 1fd32a6 | 2009-04-01 20:52:13 +0000 | [diff] [blame] | 266 | `Complementary-Multiply-with-Carry recipe |
Raymond Hettinger | 9743fd0 | 2009-04-03 05:47:33 +0000 | [diff] [blame] | 267 | <http://code.activestate.com/recipes/576707/>`_ for a compatible alternative |
| 268 | random number generator with a long period and comparatively simple update |
Raymond Hettinger | 1fd32a6 | 2009-04-01 20:52:13 +0000 | [diff] [blame] | 269 | operations. |
Raymond Hettinger | 435cb0f | 2010-09-06 23:36:31 +0000 | [diff] [blame] | 270 | |
Raymond Hettinger | 3cdf871 | 2010-12-02 05:35:35 +0000 | [diff] [blame] | 271 | |
Raymond Hettinger | 435cb0f | 2010-09-06 23:36:31 +0000 | [diff] [blame] | 272 | Notes on Reproducibility |
Antoine Pitrou | e72b586 | 2010-12-12 20:13:31 +0000 | [diff] [blame] | 273 | ------------------------ |
Raymond Hettinger | 435cb0f | 2010-09-06 23:36:31 +0000 | [diff] [blame] | 274 | |
| 275 | Sometimes it is useful to be able to reproduce the sequences given by a pseudo |
| 276 | random number generator. By re-using a seed value, the same sequence should be |
| 277 | reproducible from run to run as long as multiple threads are not running. |
| 278 | |
| 279 | Most of the random module's algorithms and seeding functions are subject to |
| 280 | change across Python versions, but two aspects are guaranteed not to change: |
| 281 | |
| 282 | * If a new seeding method is added, then a backward compatible seeder will be |
| 283 | offered. |
| 284 | |
| 285 | * The generator's :meth:`random` method will continue to produce the same |
| 286 | sequence when the compatible seeder is given the same seed. |
Raymond Hettinger | 2fdc7b1 | 2010-12-02 02:41:33 +0000 | [diff] [blame] | 287 | |
Raymond Hettinger | 6e35394 | 2010-12-04 23:42:12 +0000 | [diff] [blame] | 288 | .. _random-examples: |
Raymond Hettinger | 2fdc7b1 | 2010-12-02 02:41:33 +0000 | [diff] [blame] | 289 | |
Raymond Hettinger | 2fdc7b1 | 2010-12-02 02:41:33 +0000 | [diff] [blame] | 290 | Examples and Recipes |
Antoine Pitrou | e72b586 | 2010-12-12 20:13:31 +0000 | [diff] [blame] | 291 | -------------------- |
Raymond Hettinger | 2fdc7b1 | 2010-12-02 02:41:33 +0000 | [diff] [blame] | 292 | |
Raymond Hettinger | 3cdf871 | 2010-12-02 05:35:35 +0000 | [diff] [blame] | 293 | Basic usage:: |
| 294 | |
| 295 | >>> random.random() # Random float x, 0.0 <= x < 1.0 |
| 296 | 0.37444887175646646 |
| 297 | |
| 298 | >>> random.uniform(1, 10) # Random float x, 1.0 <= x < 10.0 |
| 299 | 1.1800146073117523 |
| 300 | |
| 301 | >>> random.randrange(10) # Integer from 0 to 9 |
| 302 | 7 |
| 303 | |
| 304 | >>> random.randrange(0, 101, 2) # Even integer from 0 to 100 |
| 305 | 26 |
| 306 | |
| 307 | >>> random.choice('abcdefghij') # Single random element |
| 308 | 'c' |
| 309 | |
| 310 | >>> items = [1, 2, 3, 4, 5, 6, 7] |
| 311 | >>> random.shuffle(items) |
| 312 | >>> items |
| 313 | [7, 3, 2, 5, 6, 4, 1] |
| 314 | |
| 315 | >>> random.sample([1, 2, 3, 4, 5], 3) # Three samples without replacement |
| 316 | [4, 1, 5] |
| 317 | |
Raymond Hettinger | 2fdc7b1 | 2010-12-02 02:41:33 +0000 | [diff] [blame] | 318 | A common task is to make a :func:`random.choice` with weighted probababilites. |
| 319 | |
| 320 | If the weights are small integer ratios, a simple technique is to build a sample |
| 321 | population with repeats:: |
| 322 | |
| 323 | >>> weighted_choices = [('Red', 3), ('Blue', 2), ('Yellow', 1), ('Green', 4)] |
| 324 | >>> population = [val for val, cnt in weighted_choices for i in range(cnt)] |
| 325 | >>> random.choice(population) |
| 326 | 'Green' |
| 327 | |
Raymond Hettinger | 3cdf871 | 2010-12-02 05:35:35 +0000 | [diff] [blame] | 328 | A more general approach is to arrange the weights in a cumulative distribution |
| 329 | with :func:`itertools.accumulate`, and then locate the random value with |
| 330 | :func:`bisect.bisect`:: |
Raymond Hettinger | 2fdc7b1 | 2010-12-02 02:41:33 +0000 | [diff] [blame] | 331 | |
| 332 | >>> choices, weights = zip(*weighted_choices) |
| 333 | >>> cumdist = list(itertools.accumulate(weights)) |
| 334 | >>> x = random.random() * cumdist[-1] |
| 335 | >>> choices[bisect.bisect(cumdist, x)] |
| 336 | 'Blue' |