четверг, 9 мая 2024 г.

Binomial distribution

from matplotlib import pyplot as plt
from math import factorial
from scipy.stats import binom

n = 30 # количество попыток
p = 0.07 # вероятность успеха

distr = []

#for k in range(0, n + 1):
#    choose = factorial(n) / (factorial(k) * factorial(n - k))
#    prob = choose * p**k * (1 - p)**(n - k) 
#    distr.append(prob)

# probability mass function — функция для дискретных распределений, показывает вероятность каждого значения
for k in range(0, n + 1):
    current_value = binom.pmf(k, n, p)
    distr.append(current_value)

plt.bar(range(0, n + 1), distr)

Комментариев нет:

Отправить комментарий