生成随机泊松数

Question

我正在编写一个函数（不使用 random.poisson 函数），它从具有 lambda a 的泊松分布中生成一个随机数。

运行不同参数的函数时，我总是得到0。我猜是算法问题，但不知道哪里错了。

这是我的代码。

#takes a as lambda (mean)
def random_poisson(a):
    x = 0
    p = 1
    while p >= math.exp(a):
        i = random.random()
        p = p * i
        x += 1

    return x

尝试不同的 lambda 值：

print(random_poisson(3))
print(random_poisson(5))
print(random_poisson(math.log(5)))

输出：

0
0
0

非常感谢！

Answer 1

对于任何正数a，math.exp(a) 大于 1，但 p 为 1。循环条件永远不会为真，循环永远不会执行，并且您的函数返回初始值 x。