blob: c690eeae191ecca9cdd8a4dc14115c0271efaff1 [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001:mod:`random` --- Generate pseudo-random numbers
2================================================
3
4.. module:: random
5 :synopsis: Generate pseudo-random numbers with various common distributions.
6
7
8This module implements pseudo-random number generators for various
9distributions.
10
11For integers, uniform selection from a range. For sequences, uniform selection
12of a random element, a function to generate a random permutation of a list
13in-place, and a function for random sampling without replacement.
14
15On the real line, there are functions to compute uniform, normal (Gaussian),
16lognormal, negative exponential, gamma, and beta distributions. For generating
17distributions of angles, the von Mises distribution is available.
18
19Almost all module functions depend on the basic function :func:`random`, which
20generates a random float uniformly in the semi-open range [0.0, 1.0). Python
21uses the Mersenne Twister as the core generator. It produces 53-bit precision
22floats and has a period of 2\*\*19937-1. The underlying implementation in C is
23both fast and threadsafe. The Mersenne Twister is one of the most extensively
24tested random number generators in existence. However, being completely
25deterministic, it is not suitable for all purposes, and is completely unsuitable
26for cryptographic purposes.
27
28The functions supplied by this module are actually bound methods of a hidden
29instance of the :class:`random.Random` class. You can instantiate your own
Raymond Hettinger28de64f2008-01-13 23:40:30 +000030instances of :class:`Random` to get generators that don't share state.
Georg Brandl116aa622007-08-15 14:28:22 +000031
32Class :class:`Random` can also be subclassed if you want to use a different
33basic generator of your own devising: in that case, override the :meth:`random`,
Raymond Hettingerafd30452009-02-24 10:57:02 +000034:meth:`seed`, :meth:`getstate`, and :meth:`setstate` methods.
Benjamin Petersond18de0e2008-07-31 20:21:46 +000035Optionally, a new generator can supply a :meth:`getrandbits` method --- this
Georg Brandl116aa622007-08-15 14:28:22 +000036allows :meth:`randrange` to produce selections over an arbitrarily large range.
37
Benjamin Peterson21896a32010-03-21 22:03:03 +000038As an example of subclassing, the :mod:`random` module provides the
39:class:`WichmannHill` class that implements an alternative generator in pure
40Python. The class provides a backward compatible way to reproduce results from
41earlier versions of Python, which used the Wichmann-Hill algorithm as the core
42generator. Note that this Wichmann-Hill generator can no longer be recommended:
43its period is too short by contemporary standards, and the sequence generated is
44known to fail some stringent randomness tests. See the references below for a
45recent variant that repairs these flaws.
46
47The :mod:`random` module also provides the :class:`SystemRandom` class which
48uses the system function :func:`os.urandom` to generate random numbers
49from sources provided by the operating system.
Georg Brandl116aa622007-08-15 14:28:22 +000050
Georg Brandl116aa622007-08-15 14:28:22 +000051Bookkeeping functions:
52
53
Raymond Hettingerf763a722010-09-07 00:38:15 +000054.. function:: seed([x], version=2)
Georg Brandl116aa622007-08-15 14:28:22 +000055
Raymond Hettingerf763a722010-09-07 00:38:15 +000056 Initialize the random number generator.
Georg Brandl116aa622007-08-15 14:28:22 +000057
Raymond Hettingerf763a722010-09-07 00:38:15 +000058 If *x* is omitted or ``None``, the current system time is used. If
59 randomness sources are provided by the operating system, they are used
60 instead of the system time (see the :func:`os.urandom` function for details
61 on availability).
Georg Brandl116aa622007-08-15 14:28:22 +000062
Raymond Hettingerf763a722010-09-07 00:38:15 +000063 If *x* is an int, it is used directly.
64
65 With version 2 (the default), a :class:`str`, :class:`bytes`, or :class:`bytearray`
66 object gets converted to a :class:`int` and all of its bits are used. With version 1,
67 the :func:`hash` of *x* is used instead.
68
69 .. versionchanged:: 3.2
70 Moved to the version 2 scheme which uses all of the bits in a string seed.
Georg Brandl116aa622007-08-15 14:28:22 +000071
72.. function:: getstate()
73
74 Return an object capturing the current internal state of the generator. This
75 object can be passed to :func:`setstate` to restore the state.
76
Georg Brandl116aa622007-08-15 14:28:22 +000077
78.. function:: setstate(state)
79
80 *state* should have been obtained from a previous call to :func:`getstate`, and
81 :func:`setstate` restores the internal state of the generator to what it was at
82 the time :func:`setstate` was called.
83
Georg Brandl116aa622007-08-15 14:28:22 +000084
Georg Brandl116aa622007-08-15 14:28:22 +000085.. function:: getrandbits(k)
86
Ezio Melotti0639d5a2009-12-19 23:26:38 +000087 Returns a Python integer with *k* random bits. This method is supplied with
Georg Brandl5c106642007-11-29 17:41:05 +000088 the MersenneTwister generator and some other generators may also provide it
Georg Brandl116aa622007-08-15 14:28:22 +000089 as an optional part of the API. When available, :meth:`getrandbits` enables
90 :meth:`randrange` to handle arbitrarily large ranges.
91
Georg Brandl116aa622007-08-15 14:28:22 +000092
93Functions for integers:
94
Georg Brandl116aa622007-08-15 14:28:22 +000095.. function:: randrange([start,] stop[, step])
96
97 Return a randomly selected element from ``range(start, stop, step)``. This is
98 equivalent to ``choice(range(start, stop, step))``, but doesn't actually build a
99 range object.
100
Raymond Hettinger05156612010-09-07 04:44:52 +0000101 The positional argument pattern matches that of :func:`range`. Keyword arguments
102 should not be used because the function may use them in unexpected ways.
103
104 .. versionchanged:: 3.2
105 :meth:`randrange` is more sophisticated about producing equally distributed
106 values. Formerly it used a style like ``int(random()*n)`` which could produce
107 slightly uneven distributions.
Georg Brandl116aa622007-08-15 14:28:22 +0000108
109.. function:: randint(a, b)
110
Raymond Hettingerafd30452009-02-24 10:57:02 +0000111 Return a random integer *N* such that ``a <= N <= b``. Alias for
112 ``randrange(a, b+1)``.
Georg Brandl116aa622007-08-15 14:28:22 +0000113
Georg Brandl116aa622007-08-15 14:28:22 +0000114
Georg Brandl55ac8f02007-09-01 13:51:09 +0000115Functions for sequences:
Georg Brandl116aa622007-08-15 14:28:22 +0000116
117.. function:: choice(seq)
118
119 Return a random element from the non-empty sequence *seq*. If *seq* is empty,
120 raises :exc:`IndexError`.
121
122
123.. function:: shuffle(x[, random])
124
125 Shuffle the sequence *x* in place. The optional argument *random* is a
126 0-argument function returning a random float in [0.0, 1.0); by default, this is
127 the function :func:`random`.
128
129 Note that for even rather small ``len(x)``, the total number of permutations of
130 *x* is larger than the period of most random number generators; this implies
131 that most permutations of a long sequence can never be generated.
132
133
134.. function:: sample(population, k)
135
Raymond Hettinger1acde192008-01-14 01:00:53 +0000136 Return a *k* length list of unique elements chosen from the population sequence
137 or set. Used for random sampling without replacement.
Georg Brandl116aa622007-08-15 14:28:22 +0000138
Georg Brandl116aa622007-08-15 14:28:22 +0000139 Returns a new list containing elements from the population while leaving the
140 original population unchanged. The resulting list is in selection order so that
141 all sub-slices will also be valid random samples. This allows raffle winners
142 (the sample) to be partitioned into grand prize and second place winners (the
143 subslices).
144
Guido van Rossum2cc30da2007-11-02 23:46:40 +0000145 Members of the population need not be :term:`hashable` or unique. If the population
Georg Brandl116aa622007-08-15 14:28:22 +0000146 contains repeats, then each occurrence is a possible selection in the sample.
147
148 To choose a sample from a range of integers, use an :func:`range` object as an
149 argument. This is especially fast and space efficient for sampling from a large
150 population: ``sample(range(10000000), 60)``.
151
152The following functions generate specific real-valued distributions. Function
153parameters are named after the corresponding variables in the distribution's
154equation, as used in common mathematical practice; most of these equations can
155be found in any statistics text.
156
157
158.. function:: random()
159
160 Return the next random floating point number in the range [0.0, 1.0).
161
162
163.. function:: uniform(a, b)
164
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000165 Return a random floating point number *N* such that ``a <= N <= b`` for
166 ``a <= b`` and ``b <= N <= a`` for ``b < a``.
Georg Brandl116aa622007-08-15 14:28:22 +0000167
Raymond Hettingerbe40db02009-06-11 23:12:14 +0000168 The end-point value ``b`` may or may not be included in the range
169 depending on floating-point rounding in the equation ``a + (b-a) * random()``.
Benjamin Peterson35e8c462008-04-24 02:34:53 +0000170
Christian Heimesfe337bf2008-03-23 21:54:12 +0000171.. function:: triangular(low, high, mode)
172
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000173 Return a random floating point number *N* such that ``low <= N <= high`` and
Christian Heimescc47b052008-03-25 14:56:36 +0000174 with the specified *mode* between those bounds. The *low* and *high* bounds
175 default to zero and one. The *mode* argument defaults to the midpoint
176 between the bounds, giving a symmetric distribution.
Christian Heimesfe337bf2008-03-23 21:54:12 +0000177
Georg Brandl116aa622007-08-15 14:28:22 +0000178
179.. function:: betavariate(alpha, beta)
180
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000181 Beta distribution. Conditions on the parameters are ``alpha > 0`` and
182 ``beta > 0``. Returned values range between 0 and 1.
Georg Brandl116aa622007-08-15 14:28:22 +0000183
184
185.. function:: expovariate(lambd)
186
Mark Dickinson2f947362009-01-07 17:54:07 +0000187 Exponential distribution. *lambd* is 1.0 divided by the desired
188 mean. It should be nonzero. (The parameter would be called
189 "lambda", but that is a reserved word in Python.) Returned values
190 range from 0 to positive infinity if *lambd* is positive, and from
191 negative infinity to 0 if *lambd* is negative.
Georg Brandl116aa622007-08-15 14:28:22 +0000192
193
194.. function:: gammavariate(alpha, beta)
195
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000196 Gamma distribution. (*Not* the gamma function!) Conditions on the
197 parameters are ``alpha > 0`` and ``beta > 0``.
Georg Brandl116aa622007-08-15 14:28:22 +0000198
199
200.. function:: gauss(mu, sigma)
201
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000202 Gaussian distribution. *mu* is the mean, and *sigma* is the standard
203 deviation. This is slightly faster than the :func:`normalvariate` function
204 defined below.
Georg Brandl116aa622007-08-15 14:28:22 +0000205
206
207.. function:: lognormvariate(mu, sigma)
208
209 Log normal distribution. If you take the natural logarithm of this
210 distribution, you'll get a normal distribution with mean *mu* and standard
211 deviation *sigma*. *mu* can have any value, and *sigma* must be greater than
212 zero.
213
214
215.. function:: normalvariate(mu, sigma)
216
217 Normal distribution. *mu* is the mean, and *sigma* is the standard deviation.
218
219
220.. function:: vonmisesvariate(mu, kappa)
221
222 *mu* is the mean angle, expressed in radians between 0 and 2\*\ *pi*, and *kappa*
223 is the concentration parameter, which must be greater than or equal to zero. If
224 *kappa* is equal to zero, this distribution reduces to a uniform random angle
225 over the range 0 to 2\*\ *pi*.
226
227
228.. function:: paretovariate(alpha)
229
230 Pareto distribution. *alpha* is the shape parameter.
231
232
233.. function:: weibullvariate(alpha, beta)
234
235 Weibull distribution. *alpha* is the scale parameter and *beta* is the shape
236 parameter.
237
238
239Alternative Generators:
240
Georg Brandl116aa622007-08-15 14:28:22 +0000241.. class:: SystemRandom([seed])
242
243 Class that uses the :func:`os.urandom` function for generating random numbers
244 from sources provided by the operating system. Not available on all systems.
245 Does not rely on software state and sequences are not reproducible. Accordingly,
Raymond Hettingerafd30452009-02-24 10:57:02 +0000246 the :meth:`seed` method has no effect and is ignored.
Georg Brandl116aa622007-08-15 14:28:22 +0000247 The :meth:`getstate` and :meth:`setstate` methods raise
248 :exc:`NotImplementedError` if called.
249
Georg Brandl116aa622007-08-15 14:28:22 +0000250
251Examples of basic usage::
252
253 >>> random.random() # Random float x, 0.0 <= x < 1.0
254 0.37444887175646646
255 >>> random.uniform(1, 10) # Random float x, 1.0 <= x < 10.0
256 1.1800146073117523
257 >>> random.randint(1, 10) # Integer from 1 to 10, endpoints included
258 7
259 >>> random.randrange(0, 101, 2) # Even integer from 0 to 100
260 26
261 >>> random.choice('abcdefghij') # Choose a random element
262 'c'
263
264 >>> items = [1, 2, 3, 4, 5, 6, 7]
265 >>> random.shuffle(items)
266 >>> items
267 [7, 3, 2, 5, 6, 4, 1]
268
269 >>> random.sample([1, 2, 3, 4, 5], 3) # Choose 3 elements
270 [4, 1, 5]
271
272
273
274.. seealso::
275
276 M. Matsumoto and T. Nishimura, "Mersenne Twister: A 623-dimensionally
277 equidistributed uniform pseudorandom number generator", ACM Transactions on
278 Modeling and Computer Simulation Vol. 8, No. 1, January pp.3-30 1998.
279
Georg Brandl116aa622007-08-15 14:28:22 +0000280
Raymond Hettinger1fd32a62009-04-01 20:52:13 +0000281 `Complementary-Multiply-with-Carry recipe
Raymond Hettinger9743fd02009-04-03 05:47:33 +0000282 <http://code.activestate.com/recipes/576707/>`_ for a compatible alternative
283 random number generator with a long period and comparatively simple update
Raymond Hettinger1fd32a62009-04-01 20:52:13 +0000284 operations.
Raymond Hettinger435cb0f2010-09-06 23:36:31 +0000285
286Notes on Reproducibility
287========================
288
289Sometimes it is useful to be able to reproduce the sequences given by a pseudo
290random number generator. By re-using a seed value, the same sequence should be
291reproducible from run to run as long as multiple threads are not running.
292
293Most of the random module's algorithms and seeding functions are subject to
294change across Python versions, but two aspects are guaranteed not to change:
295
296* If a new seeding method is added, then a backward compatible seeder will be
297 offered.
298
299* The generator's :meth:`random` method will continue to produce the same
300 sequence when the compatible seeder is given the same seed.