numpy.random.RandomState.geometric

метод

RandomState.geometric(p, size=None)

Генерирует выборки из геометрического распределения.

Эксперименты Бернулли — это эксперименты с двумя возможными исходами: успехом или неудачей (например, подбрасывание монеты). Геометрическое распределение моделирует количество испытаний, которое необходимо провести для достижения успеха. Поэтому оно определено на положительных целых числах, k = 1, 2, ....

Функция плотности вероятности геометрического распределения

где p — вероятность успеха в отдельном испытании.

Параметры:	p : float or array_like of floats Вероятность успеха в отдельном испытании. size : int or tuple of ints, optional Форма выходного массива. Если заданная форма, например, (m, n, k), то генерируется m * n * k выборок. Если размерность size отсутствует (по умолчанию), то возвращается одно значение, если p является скалярным. В противном случае генерируется np.array(p).size выборок.
Возвращает:	out : ndarray or scalar Сгенерированные выборки из геометрического распределения с заданными параметрами.

Примеры

Сгенерировать десять тысяч значений из геометрического распределения, с вероятностью успеха в отдельном испытании равной 0,35:

>>> z = np.random.geometric(p=0.35, size=10000)

Сколько испытаний потребовалось для получения успеха?

>>> (z == 1).sum() / 10000.
0.34889999999999999 #random