numpy.random.RandomState.multinomial

RandomState.multinomial(n, pvals, size=None)

Выборка из многочленного распределения.

Многочленное распределение является многомерной обобщением биномиального распределения. Рассмотрим эксперимент с одним из p возможных исходов. Примером такого эксперимента является бросание игральной кости, где результатом может быть значение от 1 до 6. Каждая выборка из распределения представляет n таких экспериментов. Ее значения, X_i = [X_0, X_1, ..., X_p], представляют количество раз, когда результат был i.

Параметры:

n : int Количество экспериментов. pvals : последовательность чисел с плавающей точкой, длина p Вероятности каждого из p различных исходов. Они должны суммироваться до 1 (однако, предполагается, что последний элемент всегда учитывает оставшуюся вероятность, при условии, что sum(pvals[:-1]) <= 1). size : int или кортеж из целых чисел, необязательно Форма выходных данных. Если заданная форма, например, (m, n, k), то m * n * k выборок генерируются. По умолчанию None, в этом случае возвращается единственное значение.

Возвращает:

out : ndarray Сгенерированные выборки, с формой size, если она была предоставлена. Если нет, форма равна (N,). Другими словами, каждый элемент out[i,j,...,:] представляет собой N-мерное значение, взятое из распределения.

Примеры

Бросание игральной кости 20 раз:

>>> np.random.multinomial(20, [1/6.]*6, size=1)
array([[4, 1, 7, 5, 2, 1]])

Она выпала 4 раза на 1, 1 раз на 2 и т.д.

Теперь бросим кость 20 раз, и ещё 20 раз:

>>> np.random.multinomial(20, [1/6.]*6, size=2)
array([[3, 4, 3, 3, 4, 3],
       [2, 4, 3, 4, 0, 7]])

Для первого набора бросков, у нас выпало 3 раза 1, 4 раза 2 и т.д. Для второго набора, у нас выпало 2 раза 1, 4 раза 2 и т.д.

Нагруженная кость с большей вероятностью выпадает на 6:

>>> np.random.multinomial(100, [1/7.]*5 + [2/7.])
array([11, 16, 14, 17, 16, 26])

Входные значения вероятностей должны быть нормированы. В качестве реализации, значение последнего элемента игнорируется и предполагается, что оно занимает оставшуюся вероятность, но на это не стоит полагаться. У нас есть смещённая монета, которая имеет в два раза больше веса на одной стороне, чем на другой, её выборка должна быть такой:

>>> np.random.multinomial(100, [1.0 / 3, 2.0 / 3])  # RIGHT
array([38, 62])

а не такой:

>>> np.random.multinomial(100, [1.0, 2.0])  # WRONG
array([100,   0])