Distribución multinomial en NumPy

La distribución multinomial en NumPy se utiliza para generar muestras de una distribución multinomial o calcular probabilidades en una distribución multinomial. Esta distribución se utiliza cuando se realizan múltiples ensayos (experimentos) independientes, y cada ensayo puede tener múltiples resultados posibles. Por ejemplo, lanzar un dado múltiples veces es un escenario donde se puede aplicar una distribución multinomialAquí tienes una lección sobre cómo usar la distribución multinomial en NumPy:


Importar NumPy

Primero, asegúrate de importar la biblioteca NumPy:

import numpy as np

Definir los parámetros para Distribución multinomial en NumPy

Debes definir los parámetros de la distribución multinomial, que incluyen el número de ensayos (n) y las probabilidades de cada resultado posible en cada ensayo. Por ejemplo, si estás lanzando un dado justo cuatro veces, y quieres calcular la probabilidad de obtener cada número del 1 al 6 en cada lanzamiento, puedes definir los parámetros de la siguiente manera:

n = 4  # Número de ensayos
p = [1/6, 1/6, 1/6, 1/6, 1/6, 1/6]  # Probabilidades de obtener 1, 2, 3, 4, 5, 6 en cada lanzamiento

Generar muestras para Distribución multinomial en NumPy

Para generar muestras de la distribución multinomial, puedes usar numpy.random.multinomial. Por ejemplo, para generar una muestra de 1000 ensayos:

sample = np.random.multinomial(n, p, size=1000)

sample será una matriz donde cada fila representa un conjunto de resultados de ensayos multinomiales.


Calcular probabilidades en la Distribución multinomial en NumPy

Si deseas calcular las probabilidades de un conjunto específico de resultados, puedes usar numpy.random.multinomial también. Por ejemplo, para calcular la probabilidad de obtener exactamente dos 1’s, tres 2’s y ningún otro número en cuatro lanzamientos:

outcome = [2, 0, 0, 3, 0, 0]  # Resultado deseado
probability = np.random.multinomial(n, p)
result = np.prod(p ** outcome) * np.math.factorial(n) / np.prod(np.math.factorial(outcome))
print(result)

Este es un ejemplo básico de cómo trabajar con la distribución multinomial en NumPy. Puedes ajustar los parámetros y las operaciones según tus necesidades específicas.