python - 如何模拟有偏硬币的翻转？

Question

在无偏掷硬币中，H 或 T 出现 50% 的次数。

但我想模拟硬币，它给出概率为“p”的 H 和概率为“(1-p)”的 T。

类似这样的:

def flip(p):
   '''this function return H with probability p'''
   # do something
   return result

>> [flip(0.8) for i in xrange(10)]
[H,H,T,H,H,H,T,H,H,H]

Answer 1

random.random() 返回 [0, 1) 范围内的均匀分布伪随机 float 。该数字小于 [0,1) 范围内的给定数字 p，概率为 p。因此:

def flip(p):
    return 'H' if random.random() < p else 'T'

一些实验:

>>> 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