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