blob: 18c063ca7baed73fb810e0f8b246dbee6480a9ec [file] [log] [blame]
Georg Brandl116aa622007-08-15 14:28:22 +00001
2:mod:`random` --- Generate pseudo-random numbers
3================================================
4
5.. module:: random
6 :synopsis: Generate pseudo-random numbers with various common distributions.
7
8
9This module implements pseudo-random number generators for various
10distributions.
11
12For integers, uniform selection from a range. For sequences, uniform selection
13of a random element, a function to generate a random permutation of a list
14in-place, and a function for random sampling without replacement.
15
16On the real line, there are functions to compute uniform, normal (Gaussian),
17lognormal, negative exponential, gamma, and beta distributions. For generating
18distributions of angles, the von Mises distribution is available.
19
20Almost all module functions depend on the basic function :func:`random`, which
21generates a random float uniformly in the semi-open range [0.0, 1.0). Python
22uses the Mersenne Twister as the core generator. It produces 53-bit precision
23floats and has a period of 2\*\*19937-1. The underlying implementation in C is
24both fast and threadsafe. The Mersenne Twister is one of the most extensively
25tested random number generators in existence. However, being completely
26deterministic, it is not suitable for all purposes, and is completely unsuitable
27for cryptographic purposes.
28
29The functions supplied by this module are actually bound methods of a hidden
30instance of the :class:`random.Random` class. You can instantiate your own
31instances of :class:`Random` to get generators that don't share state. This is
32especially useful for multi-threaded programs, creating a different instance of
33:class:`Random` for each thread, and using the :meth:`jumpahead` method to make
34it likely that the generated sequences seen by each thread don't overlap.
35
36Class :class:`Random` can also be subclassed if you want to use a different
37basic generator of your own devising: in that case, override the :meth:`random`,
38:meth:`seed`, :meth:`getstate`, :meth:`setstate` and :meth:`jumpahead` methods.
39Optionally, a new generator can supply a :meth:`getrandombits` method --- this
40allows :meth:`randrange` to produce selections over an arbitrarily large range.
41
Georg Brandl116aa622007-08-15 14:28:22 +000042As an example of subclassing, the :mod:`random` module provides the
43:class:`WichmannHill` class that implements an alternative generator in pure
44Python. The class provides a backward compatible way to reproduce results from
45earlier versions of Python, which used the Wichmann-Hill algorithm as the core
46generator. Note that this Wichmann-Hill generator can no longer be recommended:
47its period is too short by contemporary standards, and the sequence generated is
48known to fail some stringent randomness tests. See the references below for a
49recent variant that repairs these flaws.
50
Georg Brandl116aa622007-08-15 14:28:22 +000051Bookkeeping functions:
52
53
54.. function:: seed([x])
55
56 Initialize the basic random number generator. Optional argument *x* can be any
Guido van Rossum2cc30da2007-11-02 23:46:40 +000057 :term:`hashable` object. If *x* is omitted or ``None``, current system time is used;
Georg Brandl116aa622007-08-15 14:28:22 +000058 current system time is also used to initialize the generator when the module is
59 first imported. If randomness sources are provided by the operating system,
60 they are used instead of the system time (see the :func:`os.urandom` function
61 for details on availability).
62
Georg Brandl5c106642007-11-29 17:41:05 +000063 If *x* is not ``None`` or an int, ``hash(x)`` is used instead. If *x* is an
64 int, *x* is used directly.
Georg Brandl116aa622007-08-15 14:28:22 +000065
66
67.. function:: getstate()
68
69 Return an object capturing the current internal state of the generator. This
70 object can be passed to :func:`setstate` to restore the state.
71
Christian Heimescbf3b5c2007-12-03 21:02:03 +000072 State values produced in Python 2.6 cannot be loaded into earlier versions.
73
Georg Brandl116aa622007-08-15 14:28:22 +000074
75.. function:: setstate(state)
76
77 *state* should have been obtained from a previous call to :func:`getstate`, and
78 :func:`setstate` restores the internal state of the generator to what it was at
79 the time :func:`setstate` was called.
80
Georg Brandl116aa622007-08-15 14:28:22 +000081
82.. function:: jumpahead(n)
83
84 Change the internal state to one different from and likely far away from the
85 current state. *n* is a non-negative integer which is used to scramble the
86 current state vector. This is most useful in multi-threaded programs, in
87 conjuction with multiple instances of the :class:`Random` class:
88 :meth:`setstate` or :meth:`seed` can be used to force all instances into the
89 same internal state, and then :meth:`jumpahead` can be used to force the
90 instances' states far apart.
91
Georg Brandl116aa622007-08-15 14:28:22 +000092
93.. function:: getrandbits(k)
94
Georg Brandl5c106642007-11-29 17:41:05 +000095 Returns a python integer with *k* random bits. This method is supplied with
96 the MersenneTwister generator and some other generators may also provide it
Georg Brandl116aa622007-08-15 14:28:22 +000097 as an optional part of the API. When available, :meth:`getrandbits` enables
98 :meth:`randrange` to handle arbitrarily large ranges.
99
Georg Brandl116aa622007-08-15 14:28:22 +0000100
101Functions for integers:
102
Georg Brandl116aa622007-08-15 14:28:22 +0000103.. function:: randrange([start,] stop[, step])
104
105 Return a randomly selected element from ``range(start, stop, step)``. This is
106 equivalent to ``choice(range(start, stop, step))``, but doesn't actually build a
107 range object.
108
Georg Brandl116aa622007-08-15 14:28:22 +0000109
110.. function:: randint(a, b)
111
112 Return a random integer *N* such that ``a <= N <= b``.
113
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
136 Return a *k* length list of unique elements chosen from the population sequence.
137 Used for random sampling without replacement.
138
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
165 Return a random floating point number *N* such that ``a <= N < b``.
166
167
168.. function:: betavariate(alpha, beta)
169
170 Beta distribution. Conditions on the parameters are ``alpha > 0`` and ``beta >
171 0``. Returned values range between 0 and 1.
172
173
174.. function:: expovariate(lambd)
175
176 Exponential distribution. *lambd* is 1.0 divided by the desired mean. (The
177 parameter would be called "lambda", but that is a reserved word in Python.)
178 Returned values range from 0 to positive infinity.
179
180
181.. function:: gammavariate(alpha, beta)
182
183 Gamma distribution. (*Not* the gamma function!) Conditions on the parameters
184 are ``alpha > 0`` and ``beta > 0``.
185
186
187.. function:: gauss(mu, sigma)
188
189 Gaussian distribution. *mu* is the mean, and *sigma* is the standard deviation.
190 This is slightly faster than the :func:`normalvariate` function defined below.
191
192
193.. function:: lognormvariate(mu, sigma)
194
195 Log normal distribution. If you take the natural logarithm of this
196 distribution, you'll get a normal distribution with mean *mu* and standard
197 deviation *sigma*. *mu* can have any value, and *sigma* must be greater than
198 zero.
199
200
201.. function:: normalvariate(mu, sigma)
202
203 Normal distribution. *mu* is the mean, and *sigma* is the standard deviation.
204
205
206.. function:: vonmisesvariate(mu, kappa)
207
208 *mu* is the mean angle, expressed in radians between 0 and 2\*\ *pi*, and *kappa*
209 is the concentration parameter, which must be greater than or equal to zero. If
210 *kappa* is equal to zero, this distribution reduces to a uniform random angle
211 over the range 0 to 2\*\ *pi*.
212
213
214.. function:: paretovariate(alpha)
215
216 Pareto distribution. *alpha* is the shape parameter.
217
218
219.. function:: weibullvariate(alpha, beta)
220
221 Weibull distribution. *alpha* is the scale parameter and *beta* is the shape
222 parameter.
223
224
225Alternative Generators:
226
227.. class:: WichmannHill([seed])
228
229 Class that implements the Wichmann-Hill algorithm as the core generator. Has all
230 of the same methods as :class:`Random` plus the :meth:`whseed` method described
231 below. Because this class is implemented in pure Python, it is not threadsafe
232 and may require locks between calls. The period of the generator is
233 6,953,607,871,644 which is small enough to require care that two independent
234 random sequences do not overlap.
235
236
237.. function:: whseed([x])
238
239 This is obsolete, supplied for bit-level compatibility with versions of Python
240 prior to 2.1. See :func:`seed` for details. :func:`whseed` does not guarantee
241 that distinct integer arguments yield distinct internal states, and can yield no
242 more than about 2\*\*24 distinct internal states in all.
243
244
245.. class:: SystemRandom([seed])
246
247 Class that uses the :func:`os.urandom` function for generating random numbers
248 from sources provided by the operating system. Not available on all systems.
249 Does not rely on software state and sequences are not reproducible. Accordingly,
250 the :meth:`seed` and :meth:`jumpahead` methods have no effect and are ignored.
251 The :meth:`getstate` and :meth:`setstate` methods raise
252 :exc:`NotImplementedError` if called.
253
Georg Brandl116aa622007-08-15 14:28:22 +0000254
255Examples of basic usage::
256
257 >>> random.random() # Random float x, 0.0 <= x < 1.0
258 0.37444887175646646
259 >>> random.uniform(1, 10) # Random float x, 1.0 <= x < 10.0
260 1.1800146073117523
261 >>> random.randint(1, 10) # Integer from 1 to 10, endpoints included
262 7
263 >>> random.randrange(0, 101, 2) # Even integer from 0 to 100
264 26
265 >>> random.choice('abcdefghij') # Choose a random element
266 'c'
267
268 >>> items = [1, 2, 3, 4, 5, 6, 7]
269 >>> random.shuffle(items)
270 >>> items
271 [7, 3, 2, 5, 6, 4, 1]
272
273 >>> random.sample([1, 2, 3, 4, 5], 3) # Choose 3 elements
274 [4, 1, 5]
275
276
277
278.. seealso::
279
280 M. Matsumoto and T. Nishimura, "Mersenne Twister: A 623-dimensionally
281 equidistributed uniform pseudorandom number generator", ACM Transactions on
282 Modeling and Computer Simulation Vol. 8, No. 1, January pp.3-30 1998.
283
284 Wichmann, B. A. & Hill, I. D., "Algorithm AS 183: An efficient and portable
285 pseudo-random number generator", Applied Statistics 31 (1982) 188-190.
286