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

Question

我正在尝试执行Forest Fire图生成算法（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))