森林火灾生成模型中的几何分布值

Geometric distribution values in Forest Fire generation model

我正在尝试执行 (Python) 森林火灾图生成算法 (http://cs.stanford.edu/~jure/pubs/powergrowth-kdd05.pdf)。

这是伪代码的样子: 我遇到的问题是理解第二步。如何生成这些数字?

numpy 库中有一个实现 https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.geometric.html 但它不仅需要 p 值,还需要大小(在这个算法的情况下是什么大小?为什么在算法中给出几何均值,为什么不只是 p值?)

我在这里找到了一个解决方案 但我不确定它是否正确(根据此算法的需要)

有什么建议吗?

numpy.random.geometric(p / (1 - p))