blob: b1df231e736953ff7ff53873c77ebef8b95a6bff [file] [log] [blame]
Georg Brandl8ec7f652007-08-15 14:28:01 +00001:mod:`random` --- Generate pseudo-random numbers
2================================================
3
4.. module:: random
5 :synopsis: Generate pseudo-random numbers with various common distributions.
6
Éric Araujo29a0b572011-08-19 02:14:03 +02007**Source code:** :source:`Lib/random.py`
8
9--------------
Georg Brandl8ec7f652007-08-15 14:28:01 +000010
11This module implements pseudo-random number generators for various
12distributions.
13
14For integers, uniform selection from a range. For sequences, uniform selection
15of a random element, a function to generate a random permutation of a list
16in-place, and a function for random sampling without replacement.
17
18On the real line, there are functions to compute uniform, normal (Gaussian),
19lognormal, negative exponential, gamma, and beta distributions. For generating
20distributions of angles, the von Mises distribution is available.
21
22Almost all module functions depend on the basic function :func:`random`, which
23generates a random float uniformly in the semi-open range [0.0, 1.0). Python
24uses the Mersenne Twister as the core generator. It produces 53-bit precision
25floats and has a period of 2\*\*19937-1. The underlying implementation in C is
26both fast and threadsafe. The Mersenne Twister is one of the most extensively
27tested random number generators in existence. However, being completely
28deterministic, it is not suitable for all purposes, and is completely unsuitable
29for cryptographic purposes.
30
31The functions supplied by this module are actually bound methods of a hidden
32instance of the :class:`random.Random` class. You can instantiate your own
33instances of :class:`Random` to get generators that don't share state. This is
34especially useful for multi-threaded programs, creating a different instance of
35:class:`Random` for each thread, and using the :meth:`jumpahead` method to make
36it likely that the generated sequences seen by each thread don't overlap.
37
38Class :class:`Random` can also be subclassed if you want to use a different
39basic generator of your own devising: in that case, override the :meth:`random`,
40:meth:`seed`, :meth:`getstate`, :meth:`setstate` and :meth:`jumpahead` methods.
Benjamin Petersonf2eb2b42008-07-30 13:46:53 +000041Optionally, a new generator can supply a :meth:`getrandbits` method --- this
Georg Brandl8ec7f652007-08-15 14:28:01 +000042allows :meth:`randrange` to produce selections over an arbitrarily large range.
43
44.. versionadded:: 2.4
Benjamin Petersonf2eb2b42008-07-30 13:46:53 +000045 the :meth:`getrandbits` method.
Georg Brandl8ec7f652007-08-15 14:28:01 +000046
47As an example of subclassing, the :mod:`random` module provides the
48:class:`WichmannHill` class that implements an alternative generator in pure
49Python. The class provides a backward compatible way to reproduce results from
50earlier versions of Python, which used the Wichmann-Hill algorithm as the core
51generator. Note that this Wichmann-Hill generator can no longer be recommended:
52its period is too short by contemporary standards, and the sequence generated is
53known to fail some stringent randomness tests. See the references below for a
54recent variant that repairs these flaws.
55
56.. versionchanged:: 2.3
Andrew M. Kuchling25d6ddd2010-02-22 02:29:10 +000057 MersenneTwister replaced Wichmann-Hill as the default generator.
58
59The :mod:`random` module also provides the :class:`SystemRandom` class which
60uses the system function :func:`os.urandom` to generate random numbers
61from sources provided by the operating system.
Georg Brandl8ec7f652007-08-15 14:28:01 +000062
63Bookkeeping functions:
64
65
66.. function:: seed([x])
67
68 Initialize the basic random number generator. Optional argument *x* can be any
Georg Brandl7c3e79f2007-11-02 20:06:17 +000069 :term:`hashable` object. If *x* is omitted or ``None``, current system time is used;
Georg Brandl8ec7f652007-08-15 14:28:01 +000070 current system time is also used to initialize the generator when the module is
71 first imported. If randomness sources are provided by the operating system,
72 they are used instead of the system time (see the :func:`os.urandom` function
73 for details on availability).
74
75 .. versionchanged:: 2.4
76 formerly, operating system resources were not used.
77
Georg Brandl8ec7f652007-08-15 14:28:01 +000078.. function:: getstate()
79
80 Return an object capturing the current internal state of the generator. This
81 object can be passed to :func:`setstate` to restore the state.
82
83 .. versionadded:: 2.1
84
Martin v. Löwis6b449f42007-12-03 19:20:02 +000085 .. versionchanged:: 2.6
86 State values produced in Python 2.6 cannot be loaded into earlier versions.
87
Georg Brandl8ec7f652007-08-15 14:28:01 +000088
89.. function:: setstate(state)
90
91 *state* should have been obtained from a previous call to :func:`getstate`, and
92 :func:`setstate` restores the internal state of the generator to what it was at
Sandro Tosi106c2502012-08-12 15:11:58 +020093 the time :func:`getstate` was called.
Georg Brandl8ec7f652007-08-15 14:28:01 +000094
95 .. versionadded:: 2.1
96
97
98.. function:: jumpahead(n)
99
100 Change the internal state to one different from and likely far away from the
101 current state. *n* is a non-negative integer which is used to scramble the
102 current state vector. This is most useful in multi-threaded programs, in
Georg Brandl907a7202008-02-22 12:31:45 +0000103 conjunction with multiple instances of the :class:`Random` class:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000104 :meth:`setstate` or :meth:`seed` can be used to force all instances into the
105 same internal state, and then :meth:`jumpahead` can be used to force the
106 instances' states far apart.
107
108 .. versionadded:: 2.1
109
110 .. versionchanged:: 2.3
111 Instead of jumping to a specific state, *n* steps ahead, ``jumpahead(n)``
112 jumps to another state likely to be separated by many steps.
113
114
115.. function:: getrandbits(k)
116
117 Returns a python :class:`long` int with *k* random bits. This method is supplied
118 with the MersenneTwister generator and some other generators may also provide it
119 as an optional part of the API. When available, :meth:`getrandbits` enables
120 :meth:`randrange` to handle arbitrarily large ranges.
121
122 .. versionadded:: 2.4
123
124Functions for integers:
125
126
Ezio Melottied3f5902012-09-14 06:48:32 +0300127.. function:: randrange(stop)
128 randrange(start, stop[, step])
Georg Brandl8ec7f652007-08-15 14:28:01 +0000129
130 Return a randomly selected element from ``range(start, stop, step)``. This is
131 equivalent to ``choice(range(start, stop, step))``, but doesn't actually build a
132 range object.
133
134 .. versionadded:: 1.5.2
135
136
137.. function:: randint(a, b)
138
139 Return a random integer *N* such that ``a <= N <= b``.
140
141Functions for sequences:
142
143
144.. function:: choice(seq)
145
146 Return a random element from the non-empty sequence *seq*. If *seq* is empty,
147 raises :exc:`IndexError`.
148
149
150.. function:: shuffle(x[, random])
151
152 Shuffle the sequence *x* in place. The optional argument *random* is a
153 0-argument function returning a random float in [0.0, 1.0); by default, this is
154 the function :func:`random`.
155
156 Note that for even rather small ``len(x)``, the total number of permutations of
157 *x* is larger than the period of most random number generators; this implies
158 that most permutations of a long sequence can never be generated.
159
160
161.. function:: sample(population, k)
162
163 Return a *k* length list of unique elements chosen from the population sequence.
164 Used for random sampling without replacement.
165
166 .. versionadded:: 2.3
167
168 Returns a new list containing elements from the population while leaving the
169 original population unchanged. The resulting list is in selection order so that
170 all sub-slices will also be valid random samples. This allows raffle winners
171 (the sample) to be partitioned into grand prize and second place winners (the
172 subslices).
173
Georg Brandl7c3e79f2007-11-02 20:06:17 +0000174 Members of the population need not be :term:`hashable` or unique. If the population
Georg Brandl8ec7f652007-08-15 14:28:01 +0000175 contains repeats, then each occurrence is a possible selection in the sample.
176
177 To choose a sample from a range of integers, use an :func:`xrange` object as an
178 argument. This is especially fast and space efficient for sampling from a large
179 population: ``sample(xrange(10000000), 60)``.
180
181The following functions generate specific real-valued distributions. Function
182parameters are named after the corresponding variables in the distribution's
183equation, as used in common mathematical practice; most of these equations can
184be found in any statistics text.
185
186
187.. function:: random()
188
189 Return the next random floating point number in the range [0.0, 1.0).
190
191
192.. function:: uniform(a, b)
193
Georg Brandl9f7fb842009-01-18 13:24:10 +0000194 Return a random floating point number *N* such that ``a <= N <= b`` for
195 ``a <= b`` and ``b <= N <= a`` for ``b < a``.
Georg Brandlafeea072008-09-21 08:03:21 +0000196
Raymond Hettinger2c0cdca2009-06-11 23:14:53 +0000197 The end-point value ``b`` may or may not be included in the range
198 depending on floating-point rounding in the equation ``a + (b-a) * random()``.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000199
Georg Brandl21c69b12011-09-17 20:36:28 +0200200
Raymond Hettingerbbc50ea2008-03-23 13:32:32 +0000201.. function:: triangular(low, high, mode)
202
Georg Brandl9f7fb842009-01-18 13:24:10 +0000203 Return a random floating point number *N* such that ``low <= N <= high`` and
Raymond Hettingerd1452402008-03-24 06:07:49 +0000204 with the specified *mode* between those bounds. The *low* and *high* bounds
205 default to zero and one. The *mode* argument defaults to the midpoint
206 between the bounds, giving a symmetric distribution.
Raymond Hettingerc4f7bab2008-03-23 19:37:53 +0000207
Raymond Hettingerd1452402008-03-24 06:07:49 +0000208 .. versionadded:: 2.6
Raymond Hettingerc4f7bab2008-03-23 19:37:53 +0000209
Georg Brandl8ec7f652007-08-15 14:28:01 +0000210
211.. function:: betavariate(alpha, beta)
212
Georg Brandl9f7fb842009-01-18 13:24:10 +0000213 Beta distribution. Conditions on the parameters are ``alpha > 0`` and
214 ``beta > 0``. Returned values range between 0 and 1.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000215
216
217.. function:: expovariate(lambd)
218
Mark Dickinsone6dc5312009-01-07 17:48:33 +0000219 Exponential distribution. *lambd* is 1.0 divided by the desired
220 mean. It should be nonzero. (The parameter would be called
221 "lambda", but that is a reserved word in Python.) Returned values
222 range from 0 to positive infinity if *lambd* is positive, and from
223 negative infinity to 0 if *lambd* is negative.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000224
225
226.. function:: gammavariate(alpha, beta)
227
Georg Brandl9f7fb842009-01-18 13:24:10 +0000228 Gamma distribution. (*Not* the gamma function!) Conditions on the
229 parameters are ``alpha > 0`` and ``beta > 0``.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000230
Georg Brandl21c69b12011-09-17 20:36:28 +0200231 The probability distribution function is::
232
233 x ** (alpha - 1) * math.exp(-x / beta)
234 pdf(x) = --------------------------------------
235 math.gamma(alpha) * beta ** alpha
236
Georg Brandl8ec7f652007-08-15 14:28:01 +0000237
238.. function:: gauss(mu, sigma)
239
Georg Brandl9f7fb842009-01-18 13:24:10 +0000240 Gaussian distribution. *mu* is the mean, and *sigma* is the standard
241 deviation. This is slightly faster than the :func:`normalvariate` function
242 defined below.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000243
244
245.. function:: lognormvariate(mu, sigma)
246
247 Log normal distribution. If you take the natural logarithm of this
248 distribution, you'll get a normal distribution with mean *mu* and standard
249 deviation *sigma*. *mu* can have any value, and *sigma* must be greater than
250 zero.
251
252
253.. function:: normalvariate(mu, sigma)
254
255 Normal distribution. *mu* is the mean, and *sigma* is the standard deviation.
256
257
258.. function:: vonmisesvariate(mu, kappa)
259
260 *mu* is the mean angle, expressed in radians between 0 and 2\*\ *pi*, and *kappa*
261 is the concentration parameter, which must be greater than or equal to zero. If
262 *kappa* is equal to zero, this distribution reduces to a uniform random angle
263 over the range 0 to 2\*\ *pi*.
264
265
266.. function:: paretovariate(alpha)
267
268 Pareto distribution. *alpha* is the shape parameter.
269
270
271.. function:: weibullvariate(alpha, beta)
272
273 Weibull distribution. *alpha* is the scale parameter and *beta* is the shape
274 parameter.
275
276
277Alternative Generators:
278
279.. class:: WichmannHill([seed])
280
281 Class that implements the Wichmann-Hill algorithm as the core generator. Has all
282 of the same methods as :class:`Random` plus the :meth:`whseed` method described
283 below. Because this class is implemented in pure Python, it is not threadsafe
284 and may require locks between calls. The period of the generator is
285 6,953,607,871,644 which is small enough to require care that two independent
286 random sequences do not overlap.
287
288
289.. function:: whseed([x])
290
291 This is obsolete, supplied for bit-level compatibility with versions of Python
292 prior to 2.1. See :func:`seed` for details. :func:`whseed` does not guarantee
293 that distinct integer arguments yield distinct internal states, and can yield no
294 more than about 2\*\*24 distinct internal states in all.
295
296
297.. class:: SystemRandom([seed])
298
299 Class that uses the :func:`os.urandom` function for generating random numbers
300 from sources provided by the operating system. Not available on all systems.
301 Does not rely on software state and sequences are not reproducible. Accordingly,
302 the :meth:`seed` and :meth:`jumpahead` methods have no effect and are ignored.
303 The :meth:`getstate` and :meth:`setstate` methods raise
304 :exc:`NotImplementedError` if called.
305
306 .. versionadded:: 2.4
307
308Examples of basic usage::
309
310 >>> random.random() # Random float x, 0.0 <= x < 1.0
311 0.37444887175646646
312 >>> random.uniform(1, 10) # Random float x, 1.0 <= x < 10.0
313 1.1800146073117523
314 >>> random.randint(1, 10) # Integer from 1 to 10, endpoints included
315 7
316 >>> random.randrange(0, 101, 2) # Even integer from 0 to 100
317 26
318 >>> random.choice('abcdefghij') # Choose a random element
319 'c'
320
321 >>> items = [1, 2, 3, 4, 5, 6, 7]
322 >>> random.shuffle(items)
323 >>> items
324 [7, 3, 2, 5, 6, 4, 1]
325
326 >>> random.sample([1, 2, 3, 4, 5], 3) # Choose 3 elements
327 [4, 1, 5]
328
329
330
331.. seealso::
332
333 M. Matsumoto and T. Nishimura, "Mersenne Twister: A 623-dimensionally
334 equidistributed uniform pseudorandom number generator", ACM Transactions on
335 Modeling and Computer Simulation Vol. 8, No. 1, January pp.3-30 1998.
336
337 Wichmann, B. A. & Hill, I. D., "Algorithm AS 183: An efficient and portable
338 pseudo-random number generator", Applied Statistics 31 (1982) 188-190.
339
Raymond Hettingerfff2f4b2009-04-01 20:50:58 +0000340 `Complementary-Multiply-with-Carry recipe
Andrew M. Kuchlinge9d35ef2009-04-02 00:02:14 +0000341 <http://code.activestate.com/recipes/576707/>`_ for a compatible alternative
342 random number generator with a long period and comparatively simple update
Raymond Hettingerfff2f4b2009-04-01 20:50:58 +0000343 operations.