In [1]:
import matplotlib.pyplot as plt
%matplotlib inline
In [2]:
def fact(n):
if n <= 1:
return 1
else:
return n * fact(n-1)
In [3]:
def choose(n, k):
return fact(n) / (fact(n-k) * fact(k))
In [4]:
def prob_shared(n):
return 1 - choose(365, n) * fact(n) / (365**n)
In [5]:
upper_limit = 100
x = range(1, upper_limit + 1)
y = [prob_shared(xi) for xi in x]
plt.plot(x, y)
plt.plot([1, upper_limit], [0.5, 0.5])
plt.xlabel('n')
plt.ylabel('p(n)')
plt.show();
In [6]:
for i in range(1, 30):
print(i, prob_shared(i))