在Python中生成指数分布

简单的解决方案是将函数还原并预测概率,以找到数字:

def generate():
    proba = random.random()
    max = 0
    for i in range(1,M+1):
         max += (1/2)**i
         if(proba<= max):
              return i-1
    return generate()

如果M很小,则分布不收敛,因此不是所有的[0,1]都是可到达的,所以您可以再试一次。

小测试:

M = 10
res = [0 for i in range(M+1)]
for i in range(1000):
    res[generate()]+=1
plt.plot([r/1000 for r in res])