blob: 54211f4f9d346bd2b1cf2f1728824ec9f4dfff13 [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
Georg Brandl116aa622007-08-15 14:28:22 +000038
Georg Brandl116aa622007-08-15 14:28:22 +000039Bookkeeping functions:
40
41
42.. function:: seed([x])
43
44 Initialize the basic random number generator. Optional argument *x* can be any
Guido van Rossum2cc30da2007-11-02 23:46:40 +000045 :term:`hashable` object. If *x* is omitted or ``None``, current system time is used;
Georg Brandl116aa622007-08-15 14:28:22 +000046 current system time is also used to initialize the generator when the module is
47 first imported. If randomness sources are provided by the operating system,
48 they are used instead of the system time (see the :func:`os.urandom` function
49 for details on availability).
50
Georg Brandl5c106642007-11-29 17:41:05 +000051 If *x* is not ``None`` or an int, ``hash(x)`` is used instead. If *x* is an
52 int, *x* is used directly.
Georg Brandl116aa622007-08-15 14:28:22 +000053
54
55.. function:: getstate()
56
57 Return an object capturing the current internal state of the generator. This
58 object can be passed to :func:`setstate` to restore the state.
59
Georg Brandl116aa622007-08-15 14:28:22 +000060
61.. function:: setstate(state)
62
63 *state* should have been obtained from a previous call to :func:`getstate`, and
64 :func:`setstate` restores the internal state of the generator to what it was at
65 the time :func:`setstate` was called.
66
Georg Brandl116aa622007-08-15 14:28:22 +000067
Georg Brandl116aa622007-08-15 14:28:22 +000068.. function:: getrandbits(k)
69
Ezio Melotti0639d5a2009-12-19 23:26:38 +000070 Returns a Python integer with *k* random bits. This method is supplied with
Georg Brandl5c106642007-11-29 17:41:05 +000071 the MersenneTwister generator and some other generators may also provide it
Georg Brandl116aa622007-08-15 14:28:22 +000072 as an optional part of the API. When available, :meth:`getrandbits` enables
73 :meth:`randrange` to handle arbitrarily large ranges.
74
Georg Brandl116aa622007-08-15 14:28:22 +000075
76Functions for integers:
77
Georg Brandl116aa622007-08-15 14:28:22 +000078.. function:: randrange([start,] stop[, step])
79
80 Return a randomly selected element from ``range(start, stop, step)``. This is
81 equivalent to ``choice(range(start, stop, step))``, but doesn't actually build a
82 range object.
83
Georg Brandl116aa622007-08-15 14:28:22 +000084
85.. function:: randint(a, b)
86
Raymond Hettingerafd30452009-02-24 10:57:02 +000087 Return a random integer *N* such that ``a <= N <= b``. Alias for
88 ``randrange(a, b+1)``.
Georg Brandl116aa622007-08-15 14:28:22 +000089
Georg Brandl116aa622007-08-15 14:28:22 +000090
Georg Brandl55ac8f02007-09-01 13:51:09 +000091Functions for sequences:
Georg Brandl116aa622007-08-15 14:28:22 +000092
93.. function:: choice(seq)
94
95 Return a random element from the non-empty sequence *seq*. If *seq* is empty,
96 raises :exc:`IndexError`.
97
98
99.. function:: shuffle(x[, random])
100
101 Shuffle the sequence *x* in place. The optional argument *random* is a
102 0-argument function returning a random float in [0.0, 1.0); by default, this is
103 the function :func:`random`.
104
105 Note that for even rather small ``len(x)``, the total number of permutations of
106 *x* is larger than the period of most random number generators; this implies
107 that most permutations of a long sequence can never be generated.
108
109
110.. function:: sample(population, k)
111
Raymond Hettinger1acde192008-01-14 01:00:53 +0000112 Return a *k* length list of unique elements chosen from the population sequence
113 or set. Used for random sampling without replacement.
Georg Brandl116aa622007-08-15 14:28:22 +0000114
Georg Brandl116aa622007-08-15 14:28:22 +0000115 Returns a new list containing elements from the population while leaving the
116 original population unchanged. The resulting list is in selection order so that
117 all sub-slices will also be valid random samples. This allows raffle winners
118 (the sample) to be partitioned into grand prize and second place winners (the
119 subslices).
120
Guido van Rossum2cc30da2007-11-02 23:46:40 +0000121 Members of the population need not be :term:`hashable` or unique. If the population
Georg Brandl116aa622007-08-15 14:28:22 +0000122 contains repeats, then each occurrence is a possible selection in the sample.
123
124 To choose a sample from a range of integers, use an :func:`range` object as an
125 argument. This is especially fast and space efficient for sampling from a large
126 population: ``sample(range(10000000), 60)``.
127
128The following functions generate specific real-valued distributions. Function
129parameters are named after the corresponding variables in the distribution's
130equation, as used in common mathematical practice; most of these equations can
131be found in any statistics text.
132
133
134.. function:: random()
135
136 Return the next random floating point number in the range [0.0, 1.0).
137
138
139.. function:: uniform(a, b)
140
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000141 Return a random floating point number *N* such that ``a <= N <= b`` for
142 ``a <= b`` and ``b <= N <= a`` for ``b < a``.
Georg Brandl116aa622007-08-15 14:28:22 +0000143
Raymond Hettingerbe40db02009-06-11 23:12:14 +0000144 The end-point value ``b`` may or may not be included in the range
145 depending on floating-point rounding in the equation ``a + (b-a) * random()``.
Benjamin Peterson35e8c462008-04-24 02:34:53 +0000146
Christian Heimesfe337bf2008-03-23 21:54:12 +0000147.. function:: triangular(low, high, mode)
148
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000149 Return a random floating point number *N* such that ``low <= N <= high`` and
Christian Heimescc47b052008-03-25 14:56:36 +0000150 with the specified *mode* between those bounds. The *low* and *high* bounds
151 default to zero and one. The *mode* argument defaults to the midpoint
152 between the bounds, giving a symmetric distribution.
Christian Heimesfe337bf2008-03-23 21:54:12 +0000153
Georg Brandl116aa622007-08-15 14:28:22 +0000154
155.. function:: betavariate(alpha, beta)
156
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000157 Beta distribution. Conditions on the parameters are ``alpha > 0`` and
158 ``beta > 0``. Returned values range between 0 and 1.
Georg Brandl116aa622007-08-15 14:28:22 +0000159
160
161.. function:: expovariate(lambd)
162
Mark Dickinson2f947362009-01-07 17:54:07 +0000163 Exponential distribution. *lambd* is 1.0 divided by the desired
164 mean. It should be nonzero. (The parameter would be called
165 "lambda", but that is a reserved word in Python.) Returned values
166 range from 0 to positive infinity if *lambd* is positive, and from
167 negative infinity to 0 if *lambd* is negative.
Georg Brandl116aa622007-08-15 14:28:22 +0000168
169
170.. function:: gammavariate(alpha, beta)
171
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000172 Gamma distribution. (*Not* the gamma function!) Conditions on the
173 parameters are ``alpha > 0`` and ``beta > 0``.
Georg Brandl116aa622007-08-15 14:28:22 +0000174
175
176.. function:: gauss(mu, sigma)
177
Benjamin Petersonb58dda72009-01-18 22:27:04 +0000178 Gaussian distribution. *mu* is the mean, and *sigma* is the standard
179 deviation. This is slightly faster than the :func:`normalvariate` function
180 defined below.
Georg Brandl116aa622007-08-15 14:28:22 +0000181
182
183.. function:: lognormvariate(mu, sigma)
184
185 Log normal distribution. If you take the natural logarithm of this
186 distribution, you'll get a normal distribution with mean *mu* and standard
187 deviation *sigma*. *mu* can have any value, and *sigma* must be greater than
188 zero.
189
190
191.. function:: normalvariate(mu, sigma)
192
193 Normal distribution. *mu* is the mean, and *sigma* is the standard deviation.
194
195
196.. function:: vonmisesvariate(mu, kappa)
197
198 *mu* is the mean angle, expressed in radians between 0 and 2\*\ *pi*, and *kappa*
199 is the concentration parameter, which must be greater than or equal to zero. If
200 *kappa* is equal to zero, this distribution reduces to a uniform random angle
201 over the range 0 to 2\*\ *pi*.
202
203
204.. function:: paretovariate(alpha)
205
206 Pareto distribution. *alpha* is the shape parameter.
207
208
209.. function:: weibullvariate(alpha, beta)
210
211 Weibull distribution. *alpha* is the scale parameter and *beta* is the shape
212 parameter.
213
214
215Alternative Generators:
216
Georg Brandl116aa622007-08-15 14:28:22 +0000217.. class:: SystemRandom([seed])
218
219 Class that uses the :func:`os.urandom` function for generating random numbers
220 from sources provided by the operating system. Not available on all systems.
221 Does not rely on software state and sequences are not reproducible. Accordingly,
Raymond Hettingerafd30452009-02-24 10:57:02 +0000222 the :meth:`seed` method has no effect and is ignored.
Georg Brandl116aa622007-08-15 14:28:22 +0000223 The :meth:`getstate` and :meth:`setstate` methods raise
224 :exc:`NotImplementedError` if called.
225
Georg Brandl116aa622007-08-15 14:28:22 +0000226
227Examples of basic usage::
228
229 >>> random.random() # Random float x, 0.0 <= x < 1.0
230 0.37444887175646646
231 >>> random.uniform(1, 10) # Random float x, 1.0 <= x < 10.0
232 1.1800146073117523
233 >>> random.randint(1, 10) # Integer from 1 to 10, endpoints included
234 7
235 >>> random.randrange(0, 101, 2) # Even integer from 0 to 100
236 26
237 >>> random.choice('abcdefghij') # Choose a random element
238 'c'
239
240 >>> items = [1, 2, 3, 4, 5, 6, 7]
241 >>> random.shuffle(items)
242 >>> items
243 [7, 3, 2, 5, 6, 4, 1]
244
245 >>> random.sample([1, 2, 3, 4, 5], 3) # Choose 3 elements
246 [4, 1, 5]
247
248
249
250.. seealso::
251
252 M. Matsumoto and T. Nishimura, "Mersenne Twister: A 623-dimensionally
253 equidistributed uniform pseudorandom number generator", ACM Transactions on
254 Modeling and Computer Simulation Vol. 8, No. 1, January pp.3-30 1998.
255
Georg Brandl116aa622007-08-15 14:28:22 +0000256
Raymond Hettinger1fd32a62009-04-01 20:52:13 +0000257 `Complementary-Multiply-with-Carry recipe
Raymond Hettinger9743fd02009-04-03 05:47:33 +0000258 <http://code.activestate.com/recipes/576707/>`_ for a compatible alternative
259 random number generator with a long period and comparatively simple update
Raymond Hettinger1fd32a62009-04-01 20:52:13 +0000260 operations.