blob: 1316d2d56906837f996b8804feea3b81f8bf54cc [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
Georg Brandld6dd0e62016-02-19 08:57:23 +010022Almost all module functions depend on the basic function :func:`.random`, which
Georg Brandl8ec7f652007-08-15 14:28:01 +000023generates 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
Georg Brandld6dd0e62016-02-19 08:57:23 +010039basic generator of your own devising: in that case, override the :meth:`~Random.random`,
40:meth:`~Random.seed`, :meth:`~Random.getstate`, :meth:`~Random.setstate` and
41:meth:`~Random.jumpahead` methods. Optionally, a new generator can supply a
42:meth:`~Random.getrandbits` method --- this
Georg Brandl8ec7f652007-08-15 14:28:01 +000043allows :meth:`randrange` to produce selections over an arbitrarily large range.
44
45.. versionadded:: 2.4
Benjamin Petersonf2eb2b42008-07-30 13:46:53 +000046 the :meth:`getrandbits` method.
Georg Brandl8ec7f652007-08-15 14:28:01 +000047
48As an example of subclassing, the :mod:`random` module provides the
49:class:`WichmannHill` class that implements an alternative generator in pure
50Python. The class provides a backward compatible way to reproduce results from
51earlier versions of Python, which used the Wichmann-Hill algorithm as the core
52generator. Note that this Wichmann-Hill generator can no longer be recommended:
53its period is too short by contemporary standards, and the sequence generated is
54known to fail some stringent randomness tests. See the references below for a
55recent variant that repairs these flaws.
56
57.. versionchanged:: 2.3
Andrew M. Kuchling25d6ddd2010-02-22 02:29:10 +000058 MersenneTwister replaced Wichmann-Hill as the default generator.
59
60The :mod:`random` module also provides the :class:`SystemRandom` class which
61uses the system function :func:`os.urandom` to generate random numbers
62from sources provided by the operating system.
Georg Brandl8ec7f652007-08-15 14:28:01 +000063
Antoine Pitrou79a16a82013-08-16 19:19:40 +020064.. warning::
65
66 The pseudo-random generators of this module should not be used for
67 security purposes. Use :func:`os.urandom` or :class:`SystemRandom` if
68 you require a cryptographically secure pseudo-random number generator.
69
70
Georg Brandl8ec7f652007-08-15 14:28:01 +000071Bookkeeping functions:
72
73
74.. function:: seed([x])
75
76 Initialize the basic random number generator. Optional argument *x* can be any
Georg Brandl7c3e79f2007-11-02 20:06:17 +000077 :term:`hashable` object. If *x* is omitted or ``None``, current system time is used;
Georg Brandl8ec7f652007-08-15 14:28:01 +000078 current system time is also used to initialize the generator when the module is
79 first imported. If randomness sources are provided by the operating system,
80 they are used instead of the system time (see the :func:`os.urandom` function
81 for details on availability).
82
83 .. versionchanged:: 2.4
84 formerly, operating system resources were not used.
85
Georg Brandl8ec7f652007-08-15 14:28:01 +000086.. function:: getstate()
87
88 Return an object capturing the current internal state of the generator. This
89 object can be passed to :func:`setstate` to restore the state.
90
91 .. versionadded:: 2.1
92
Martin v. Löwis6b449f42007-12-03 19:20:02 +000093 .. versionchanged:: 2.6
94 State values produced in Python 2.6 cannot be loaded into earlier versions.
95
Georg Brandl8ec7f652007-08-15 14:28:01 +000096
97.. function:: setstate(state)
98
99 *state* should have been obtained from a previous call to :func:`getstate`, and
100 :func:`setstate` restores the internal state of the generator to what it was at
Sandro Tosi106c2502012-08-12 15:11:58 +0200101 the time :func:`getstate` was called.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000102
103 .. versionadded:: 2.1
104
105
106.. function:: jumpahead(n)
107
108 Change the internal state to one different from and likely far away from the
109 current state. *n* is a non-negative integer which is used to scramble the
110 current state vector. This is most useful in multi-threaded programs, in
Georg Brandl907a7202008-02-22 12:31:45 +0000111 conjunction with multiple instances of the :class:`Random` class:
Georg Brandl8ec7f652007-08-15 14:28:01 +0000112 :meth:`setstate` or :meth:`seed` can be used to force all instances into the
113 same internal state, and then :meth:`jumpahead` can be used to force the
114 instances' states far apart.
115
116 .. versionadded:: 2.1
117
118 .. versionchanged:: 2.3
119 Instead of jumping to a specific state, *n* steps ahead, ``jumpahead(n)``
120 jumps to another state likely to be separated by many steps.
121
122
123.. function:: getrandbits(k)
124
125 Returns a python :class:`long` int with *k* random bits. This method is supplied
126 with the MersenneTwister generator and some other generators may also provide it
127 as an optional part of the API. When available, :meth:`getrandbits` enables
128 :meth:`randrange` to handle arbitrarily large ranges.
129
130 .. versionadded:: 2.4
131
132Functions for integers:
133
134
Ezio Melottied3f5902012-09-14 06:48:32 +0300135.. function:: randrange(stop)
136 randrange(start, stop[, step])
Georg Brandl8ec7f652007-08-15 14:28:01 +0000137
138 Return a randomly selected element from ``range(start, stop, step)``. This is
139 equivalent to ``choice(range(start, stop, step))``, but doesn't actually build a
140 range object.
141
142 .. versionadded:: 1.5.2
143
144
145.. function:: randint(a, b)
146
147 Return a random integer *N* such that ``a <= N <= b``.
148
149Functions for sequences:
150
151
152.. function:: choice(seq)
153
154 Return a random element from the non-empty sequence *seq*. If *seq* is empty,
155 raises :exc:`IndexError`.
156
157
158.. function:: shuffle(x[, random])
159
160 Shuffle the sequence *x* in place. The optional argument *random* is a
161 0-argument function returning a random float in [0.0, 1.0); by default, this is
Georg Brandld6dd0e62016-02-19 08:57:23 +0100162 the function :func:`.random`.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000163
164 Note that for even rather small ``len(x)``, the total number of permutations of
165 *x* is larger than the period of most random number generators; this implies
166 that most permutations of a long sequence can never be generated.
167
168
169.. function:: sample(population, k)
170
171 Return a *k* length list of unique elements chosen from the population sequence.
172 Used for random sampling without replacement.
173
174 .. versionadded:: 2.3
175
176 Returns a new list containing elements from the population while leaving the
177 original population unchanged. The resulting list is in selection order so that
178 all sub-slices will also be valid random samples. This allows raffle winners
179 (the sample) to be partitioned into grand prize and second place winners (the
180 subslices).
181
Georg Brandl7c3e79f2007-11-02 20:06:17 +0000182 Members of the population need not be :term:`hashable` or unique. If the population
Georg Brandl8ec7f652007-08-15 14:28:01 +0000183 contains repeats, then each occurrence is a possible selection in the sample.
184
185 To choose a sample from a range of integers, use an :func:`xrange` object as an
186 argument. This is especially fast and space efficient for sampling from a large
187 population: ``sample(xrange(10000000), 60)``.
188
189The following functions generate specific real-valued distributions. Function
190parameters are named after the corresponding variables in the distribution's
191equation, as used in common mathematical practice; most of these equations can
192be found in any statistics text.
193
194
195.. function:: random()
196
197 Return the next random floating point number in the range [0.0, 1.0).
198
199
200.. function:: uniform(a, b)
201
Georg Brandl9f7fb842009-01-18 13:24:10 +0000202 Return a random floating point number *N* such that ``a <= N <= b`` for
203 ``a <= b`` and ``b <= N <= a`` for ``b < a``.
Georg Brandlafeea072008-09-21 08:03:21 +0000204
Raymond Hettinger2c0cdca2009-06-11 23:14:53 +0000205 The end-point value ``b`` may or may not be included in the range
206 depending on floating-point rounding in the equation ``a + (b-a) * random()``.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000207
Georg Brandl21c69b12011-09-17 20:36:28 +0200208
Raymond Hettingerbbc50ea2008-03-23 13:32:32 +0000209.. function:: triangular(low, high, mode)
210
Georg Brandl9f7fb842009-01-18 13:24:10 +0000211 Return a random floating point number *N* such that ``low <= N <= high`` and
Raymond Hettingerd1452402008-03-24 06:07:49 +0000212 with the specified *mode* between those bounds. The *low* and *high* bounds
213 default to zero and one. The *mode* argument defaults to the midpoint
214 between the bounds, giving a symmetric distribution.
Raymond Hettingerc4f7bab2008-03-23 19:37:53 +0000215
Raymond Hettingerd1452402008-03-24 06:07:49 +0000216 .. versionadded:: 2.6
Raymond Hettingerc4f7bab2008-03-23 19:37:53 +0000217
Georg Brandl8ec7f652007-08-15 14:28:01 +0000218
219.. function:: betavariate(alpha, beta)
220
Georg Brandl9f7fb842009-01-18 13:24:10 +0000221 Beta distribution. Conditions on the parameters are ``alpha > 0`` and
222 ``beta > 0``. Returned values range between 0 and 1.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000223
224
225.. function:: expovariate(lambd)
226
Mark Dickinsone6dc5312009-01-07 17:48:33 +0000227 Exponential distribution. *lambd* is 1.0 divided by the desired
228 mean. It should be nonzero. (The parameter would be called
229 "lambda", but that is a reserved word in Python.) Returned values
230 range from 0 to positive infinity if *lambd* is positive, and from
231 negative infinity to 0 if *lambd* is negative.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000232
233
234.. function:: gammavariate(alpha, beta)
235
Georg Brandl9f7fb842009-01-18 13:24:10 +0000236 Gamma distribution. (*Not* the gamma function!) Conditions on the
237 parameters are ``alpha > 0`` and ``beta > 0``.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000238
Georg Brandl21c69b12011-09-17 20:36:28 +0200239 The probability distribution function is::
240
241 x ** (alpha - 1) * math.exp(-x / beta)
242 pdf(x) = --------------------------------------
243 math.gamma(alpha) * beta ** alpha
244
Georg Brandl8ec7f652007-08-15 14:28:01 +0000245
246.. function:: gauss(mu, sigma)
247
Georg Brandl9f7fb842009-01-18 13:24:10 +0000248 Gaussian distribution. *mu* is the mean, and *sigma* is the standard
249 deviation. This is slightly faster than the :func:`normalvariate` function
250 defined below.
Georg Brandl8ec7f652007-08-15 14:28:01 +0000251
252
253.. function:: lognormvariate(mu, sigma)
254
255 Log normal distribution. If you take the natural logarithm of this
256 distribution, you'll get a normal distribution with mean *mu* and standard
257 deviation *sigma*. *mu* can have any value, and *sigma* must be greater than
258 zero.
259
260
261.. function:: normalvariate(mu, sigma)
262
263 Normal distribution. *mu* is the mean, and *sigma* is the standard deviation.
264
265
266.. function:: vonmisesvariate(mu, kappa)
267
268 *mu* is the mean angle, expressed in radians between 0 and 2\*\ *pi*, and *kappa*
269 is the concentration parameter, which must be greater than or equal to zero. If
270 *kappa* is equal to zero, this distribution reduces to a uniform random angle
271 over the range 0 to 2\*\ *pi*.
272
273
274.. function:: paretovariate(alpha)
275
276 Pareto distribution. *alpha* is the shape parameter.
277
278
279.. function:: weibullvariate(alpha, beta)
280
281 Weibull distribution. *alpha* is the scale parameter and *beta* is the shape
282 parameter.
283
284
285Alternative Generators:
286
287.. class:: WichmannHill([seed])
288
289 Class that implements the Wichmann-Hill algorithm as the core generator. Has all
290 of the same methods as :class:`Random` plus the :meth:`whseed` method described
291 below. Because this class is implemented in pure Python, it is not threadsafe
292 and may require locks between calls. The period of the generator is
293 6,953,607,871,644 which is small enough to require care that two independent
294 random sequences do not overlap.
295
296
297.. function:: whseed([x])
298
299 This is obsolete, supplied for bit-level compatibility with versions of Python
300 prior to 2.1. See :func:`seed` for details. :func:`whseed` does not guarantee
301 that distinct integer arguments yield distinct internal states, and can yield no
302 more than about 2\*\*24 distinct internal states in all.
303
304
305.. class:: SystemRandom([seed])
306
307 Class that uses the :func:`os.urandom` function for generating random numbers
308 from sources provided by the operating system. Not available on all systems.
309 Does not rely on software state and sequences are not reproducible. Accordingly,
310 the :meth:`seed` and :meth:`jumpahead` methods have no effect and are ignored.
311 The :meth:`getstate` and :meth:`setstate` methods raise
312 :exc:`NotImplementedError` if called.
313
314 .. versionadded:: 2.4
315
316Examples of basic usage::
317
318 >>> random.random() # Random float x, 0.0 <= x < 1.0
319 0.37444887175646646
320 >>> random.uniform(1, 10) # Random float x, 1.0 <= x < 10.0
321 1.1800146073117523
322 >>> random.randint(1, 10) # Integer from 1 to 10, endpoints included
323 7
324 >>> random.randrange(0, 101, 2) # Even integer from 0 to 100
325 26
326 >>> random.choice('abcdefghij') # Choose a random element
327 'c'
328
329 >>> items = [1, 2, 3, 4, 5, 6, 7]
330 >>> random.shuffle(items)
331 >>> items
332 [7, 3, 2, 5, 6, 4, 1]
333
334 >>> random.sample([1, 2, 3, 4, 5], 3) # Choose 3 elements
335 [4, 1, 5]
336
337
338
339.. seealso::
340
341 M. Matsumoto and T. Nishimura, "Mersenne Twister: A 623-dimensionally
342 equidistributed uniform pseudorandom number generator", ACM Transactions on
343 Modeling and Computer Simulation Vol. 8, No. 1, January pp.3-30 1998.
344
345 Wichmann, B. A. & Hill, I. D., "Algorithm AS 183: An efficient and portable
346 pseudo-random number generator", Applied Statistics 31 (1982) 188-190.
347
Raymond Hettingerfff2f4b2009-04-01 20:50:58 +0000348 `Complementary-Multiply-with-Carry recipe
Andrew M. Kuchlinge9d35ef2009-04-02 00:02:14 +0000349 <http://code.activestate.com/recipes/576707/>`_ for a compatible alternative
350 random number generator with a long period and comparatively simple update
Raymond Hettingerfff2f4b2009-04-01 20:50:58 +0000351 operations.