#Cálculo das probabilidades para o paradoxo do aniversariante
#Veja: https://pt.wikipedia.org/wiki/Paradoxo_do_anivers%C3%A1rio

import matplotlib.pyplot as plt

probabilidades = [0,]
p = 1
n_dias = 60 # número de dias a calcular

for i in range(0, n_dias):
      p *= (1 - i/365)   # Anos bissextos são desprezados 
      probabilidades.append(1 - p)


# Plota o gráfico
fig, ax = plt.subplots(1, 1)

plt.step(range(0, n_dias+1), probabilidades)

# Definições de eixos
ax.xaxis.set_ticks_position('bottom')
ax.set_ylim(bottom=0, top=1)
ax.set_xlim(left=0, right=n_dias)


# Adiciona rótulo aos eixos
plt.xlabel('número de pessoas', fontsize=14)
plt.ylabel('probabilidade de pareamento', fontsize=14)

# Adiciona grelha
plt.grid(True)

# Adiciona linha de ~50%
h = probabilidades[23] 
plt.plot([0, 23], [h, h], color='red', linestyle='dotted', linewidth=2)
plt.plot([23, 23], [0, h], color='red', linestyle='dotted', linewidth=2)

# Escreve p(23) = 0.507
ax.text(24, 0.25, 'p(23) = {:.3f}'.format(h), horizontalalignment='left', fontsize=14)

plt.savefig('probabilidade.svg', format='svg')