Random sampling (numpy.random)¶
Numpy’s random number routines produce pseudo random numbers using
combinations of a BitGenerator to create sequences and a Generator
to use those sequences to sample from different statistical distributions:
BitGenerators: Objects that generate random numbers. These are typically unsigned integer words filled with sequences of either 32 or 64 random bits.
Generators: Objects that transform sequences of random bits from a BitGenerator into sequences of numbers that follow a specific probability distribution (such as uniform, Normal or Binomial) within a specified interval.
Since Numpy version 1.17.0 the Generator can be initialized with a
number of different BitGenerators. It exposes many different probability
distributions. See NEP 19 for context on the updated random Numpy number
routines. The legacy RandomState random number routines are still
available, but limited to a single BitGenerator. See What’s New or Different
for a complete list of improvements and differences from the legacy
RandomState.
For convenience and backward compatibility, a single RandomState
instance’s methods are imported into the numpy.random namespace, see
Legacy Random Generation for the complete list.
Quick Start¶
Call default_rng to get a new instance of a Generator, then call its
methods to obtain samples from different distributions. By default,
Generator uses bits provided by PCG64 which has better statistical
properties than the legacy MT19937 used in RandomState.
# Do this (new version)
from numpy.random import default_rng
rng = default_rng()
vals = rng.standard_normal(10)
more_vals = rng.standard_normal(10)
# instead of this (legacy version)
from numpy import random
vals = random.standard_normal(10)
more_vals = random.standard_normal(10)
Generator can be used as a replacement for RandomState. Both class
instances hold an internal BitGenerator instance to provide the bit
stream, it is accessible as gen.bit_generator. Some long-overdue API
cleanup means that legacy and compatibility methods have been removed from
Generator
Notes |
||
|
|
Compatible with |
|
||
|
|
Add an |
|
||
|
removed |
Use |
|
removed |
See What’s New or Different for more information.
Something like the following code can be used to support both RandomState
and Generator, with the understanding that the interfaces are slightly
different
try:
rng_integers = rng.integers
except AttributeError:
rng_integers = rng.randint
a = rng_integers(1000)
Seeds can be passed to any of the BitGenerators. The provided value is mixed
via SeedSequence to spread a possible sequence of seeds across a wider
range of initialization states for the BitGenerator. Here PCG64 is used and
is wrapped with a Generator.
from numpy.random import Generator, PCG64
rng = Generator(PCG64(12345))
rng.standard_normal()
Here we use default_rng to create an instance of Generator to generate a
random float:
>>> import numpy as np
>>> rng = np.random.default_rng(12345)
>>> print(rng)
Generator(PCG64)
>>> rfloat = rng.random()
>>> rfloat
0.22733602246716966
>>> type(rfloat)
<class 'float'>
Here we use default_rng to create an instance of Generator to generate 3
random integers between 0 (inclusive) and 10 (exclusive):
>>> import numpy as np
>>> rng = np.random.default_rng(12345)
>>> rints = rng.integers(low=0, high=10, size=3)
>>> rints
array([6, 2, 7])
>>> type(rints[0])
<class 'numpy.int64'>
Introduction¶
The new infrastructure takes a different approach to producing random numbers
from the RandomState object. Random number generation is separated into
two components, a bit generator and a random generator.
The BitGenerator has a limited set of responsibilities. It manages state
and provides functions to produce random doubles and random unsigned 32- and
64-bit values.
The random generator takes the
bit generator-provided stream and transforms them into more useful
distributions, e.g., simulated normal random values. This structure allows
alternative bit generators to be used with little code duplication.
The Generator is the user-facing object that is nearly identical to the
legacy RandomState. It accepts a bit generator instance as an argument.
The default is currently PCG64 but this may change in future versions.
As a convenience NumPy provides the default_rng function to hide these
details:
>>> from numpy.random import default_rng
>>> rng = default_rng(12345)
>>> print(rng)
Generator(PCG64)
>>> print(rng.random())
0.22733602246716966
One can also instantiate Generator directly with a BitGenerator instance.
To use the default PCG64 bit generator, one can instantiate it directly and
pass it to Generator:
>>> from numpy.random import Generator, PCG64
>>> rng = Generator(PCG64(12345))
>>> print(rng)
Generator(PCG64)
Similarly to use the older MT19937 bit generator (not recommended), one can
instantiate it directly and pass it to Generator:
>>> from numpy.random import Generator, MT19937
>>> rng = Generator(MT19937(12345))
>>> print(rng)
Generator(MT19937)
What’s New or Different¶
Warning
The Box-Muller method used to produce NumPy’s normals is no longer available
in Generator. It is not possible to reproduce the exact random
values using Generator for the normal distribution or any other
distribution that relies on the normal such as the RandomState.gamma or
RandomState.standard_t. If you require bitwise backward compatible
streams, use RandomState.
The Generator’s normal, exponential and gamma functions use 256-step Ziggurat methods which are 2-10 times faster than NumPy’s Box-Muller or inverse CDF implementations.
Optional
dtypeargument that acceptsnp.float32ornp.float64to produce either single or double prevision uniform random variables for select distributionsOptional
outargument that allows existing arrays to be filled for select distributionsAll BitGenerators can produce doubles, uint64s and uint32s via CTypes (
PCG64.ctypes) and CFFI (PCG64.cffi). This allows the bit generators to be used in numba.The bit generators can be used in downstream projects via Cython.
Generator.integersis now the canonical way to generate integer random numbers from a discrete uniform distribution. Therandandrandnmethods are only available through the legacyRandomState. Theendpointkeyword can be used to specify open or closed intervals. This replaces bothrandintand the deprecatedrandom_integers.Generator.randomis now the canonical way to generate floating-point random numbers, which replacesRandomState.random_sample, RandomState.sample, and RandomState.ranf. This is consistent with Python’srandom.random.All BitGenerators in numpy use
SeedSequenceto convert seeds into initialized states.The addition of an
axiskeyword argument to methods such asGenerator.choice,Generator.permutation, andGenerator.shuffleimproves support for sampling from and shuffling multi-dimensional arrays.
See What’s New or Different for a complete list of improvements and
differences from the traditional Randomstate.
Parallel Generation¶
The included generators can be used in parallel, distributed applications in one of three ways:
Users with a very large amount of parallelism will want to consult Upgrading PCG64 with PCG64DXSM.
Concepts¶
Features¶
Original Source of the Generator and BitGenerators¶
This package was developed independently of NumPy and was integrated in version 1.17.0. The original repo is at https://github.com/bashtage/randomgen.