mirror of
https://github.com/TheAlgorithms/Python.git
synced 2025-01-18 16:27:02 +00:00
Added a binomial distribution formula calculator algorithm (#2197)
* Add files via upload * Update binomial_distribution.py * Update maths/binomial_distribution.py Co-authored-by: Christian Clauss <cclauss@me.com> * Update binomial_distribution.py * Update maths/binomial_distribution.py Co-authored-by: Christian Clauss <cclauss@me.com> * Update binomial_distribution.py * Update binomial_distribution.py * Update binomial_distribution.py * Update binomial_distribution.py * Update binomial_distribution.py Co-authored-by: Christian Clauss <cclauss@me.com>
This commit is contained in:
parent
e292ddb5ec
commit
749ffd8c6f
40
maths/binomial_distribution.py
Normal file
40
maths/binomial_distribution.py
Normal file
|
@ -0,0 +1,40 @@
|
|||
"""For more information about the Binomial Distribution -
|
||||
https://en.wikipedia.org/wiki/Binomial_distribution"""
|
||||
from math import factorial
|
||||
|
||||
|
||||
def binomial_distribution(successes: int, trials: int, prob: float) -> float:
|
||||
"""
|
||||
Return probability of k successes out of n tries, with p probability for one
|
||||
success
|
||||
|
||||
The function uses the factorial function in order to calculate the binomial
|
||||
coefficient
|
||||
|
||||
>>> binomial_distribution(3, 5, 0.7)
|
||||
0.30870000000000003
|
||||
>>> binomial_distribution (2, 4, 0.5)
|
||||
0.375
|
||||
"""
|
||||
if successes > trials:
|
||||
raise ValueError("""successes must be lower or equal to trials""")
|
||||
if trials < 0 or successes < 0:
|
||||
raise ValueError("the function is defined for non-negative integers")
|
||||
if not isinstance(successes, int) or not isinstance(trials, int):
|
||||
raise ValueError("the function is defined for non-negative integers")
|
||||
if not 0 < prob < 1:
|
||||
raise ValueError("prob has to be in range of 1 - 0")
|
||||
probability = (prob ** successes) * ((1 - prob) ** (trials - successes))
|
||||
# Calculate the binomial coefficient: n! / k!(n-k)!
|
||||
coefficient = float(factorial(trials))
|
||||
coefficient /= factorial(successes) * factorial(trials - successes)
|
||||
return probability * coefficient
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
from doctest import testmod
|
||||
|
||||
testmod()
|
||||
print("Probability of 2 successes out of 4 trails")
|
||||
print("with probability of 0.75 is:", end=" ")
|
||||
print(binomial_distribution(2, 4, 0.75))
|
Loading…
Reference in New Issue
Block a user