Значения геометрического распределения в модели генерации лесного пожара
Я пытаюсь сделать реализацию (Python) алгоритма генерации графа Forest Fire ( http://cs.stanford.edu/~jure/pubs/powergrowth-kdd05.pdf).
Вот как выглядит псевдокод: 
У меня проблема с пониманием второго шага. Как сгенерировать эти числа?
В библиотеке numpy есть реализация https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.geometric.html но она требует не только значения p, но и размера (размер которого в случае этого Алгоритм? Почему в алгоритмах даны геометрические средства, а не просто значение p?)
Я нашел одно решение здесь /questions/21597401/python-ozhidal-blok-s-otstupom/21597415#21597415 но я не уверен, что оно работает правильно (в соответствии с потребностями этого алгоритма)
Какие-либо предложения?