How do I simulate flip of biased coin in python?
random.random()
returns a uniformly distributed pseudo-random floating point number in the range [0, 1). This number is less than a given number p
in the range [0,1) with probability p
. Thus:
def flip(p):
return 'H' if random.random() < p else 'T'
Some experiments:
>>> N = 100
>>> flips = [flip(0.2) for i in xrange(N)]
>>> float(flips.count('H'))/N
0.17999999999999999 # Approximately 20% of the coins are heads
>>> N = 10000
>>> flips = [flip(0.2) for i in xrange(N)]
>>> float(flips.count('H'))/N
0.20549999999999999 # Better approximation
Do you want the "bias" to be based in symmetric distribuition? Or maybe exponential distribution? Gaussian anyone?
Well, here are all the methods, extracted from random documentation itself.
First, an example of triangular distribution:
print random.triangular(0, 1, 0.7)
random.triangular(low, high, mode)
:Return a random floating point number
N
such thatlow <= N < high
and with the specified mode between those bounds. Thelow
andhigh
bounds default to zero and one. Themode
argument defaults to the midpoint between the bounds, giving a symmetric distribution.
random.betavariate(alpha, beta)
:Beta distribution. Conditions on the parameters are
alpha > 0
andbeta > 0
. Returned values range between0
and1
.
random.expovariate(lambd)
:Exponential distribution.
lambd
is1.0
divided by the desired mean. It should be nonzero. (The parameter would be called “lambda
”, but that is a reserved word in Python.) Returned values range from0
to positive infinity iflambd
is positive, and from negative infinity to0
iflambd
is negative.
random.gammavariate(alpha, beta)
:Gamma distribution. (Not the gamma function!) Conditions on the parameters are
alpha > 0
andbeta > 0
.
random.gauss(mu, sigma)
:Gaussian distribution.
mu
is the mean, andsigma
is the standard deviation. This is slightly faster than thenormalvariate()
function defined below.
random.lognormvariate(mu, sigma)
:Log normal distribution. If you take the natural logarithm of this distribution, you’ll get a normal distribution with mean
mu
and standard deviationsigma
.mu
can have any value, andsigma
must be greater than zero.
random.normalvariate(mu, sigma)
:Normal distribution.
mu
is the mean, andsigma
is the standard deviation.
random.vonmisesvariate(mu, kappa)
:
mu
is the mean angle, expressed in radians between0
and2*pi
, andkappa
is the concentration parameter, which must be greater than or equal to zero. Ifkappa
is equal to zero, this distribution reduces to a uniform random angle over the range0
to2*pi
.
random.paretovariate(alpha)
:Pareto distribution.
alpha
is the shape parameter.
random.weibullvariate(alpha, beta)
Weibull distribution.
alpha
is the scale parameter andbeta
is the shape parameter.
import random
def flip(p):
return (random.random() < p)
That returns a boolean which you can then use to choose H or T (or choose between any two values) you want. You could also include the choice in the method:
def flip(p):
if random.random() < p:
return 'H'
else:
return 'T'
but it'd be less generally useful that way.