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

Question

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

下面是伪代码的样子：

我遇到的问题是理解第二步。如何生成这些数字？

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

我在这里找到了一个解决方案，https://stackoverflow.com/a/9083170/7127824，但我不确定它是否正确(根据这个算法的需要)

有什么建议吗？

Answer 1

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