import numpy as np
import matplotlib.pyplot as plt

# Hvor mange dager?
n = 365

# Ulike verdier av r
rMax = 100
rVec = np.arange(1, rMax+1, 1, dtype=int)

# Regn ut sannsynligheter for at alle har bursdag forskjellige dager
pTmp = np.ones(rMax)
for i in np.arange(1, rMax):
	pTmp[i] = pTmp[i-1]*(n-i)/n

# Regn ut komplementhendelsen. Minst to har bursdag samme dag
pVec = 1 - pTmp

# Lag plott
plt.plot(rVec, pVec, 'o-')
plt.xlabel('Personer')
plt.ylabel('Sannsynlighet')
#plt.show()

# Plott linjer
x1 = [23, 23]
y1 = [0, pVec[23-1]]
plt.plot(x1, y1, 'g-')
x1 = [0, 23]
y1 = [pVec[23-1], pVec[23-1]]
plt.plot(x1, y1, 'g-')
plt.show()