# -*- coding: utf-8 -*-
"""
Created on Wed Feb 25 09:52:21 2015

@author: utilisateur
"""

import random as rd
import numpy as np
import matplotlib.pyplot as plt



def simul_X(L):
    tir = rd.random()
    s = 0
    for i in range(len(L)):
        if s < tir and ... :
            return ...
        s = s+L[i]    #

#test 
L1 = [0.3,0.5,0.2] 
L2 = [0.2,0.5,0.3]

print(simul_X(L1))      
print(simul_X(L2))    

def simul_N(n,L):
    N = 1
    k = 0
    while k ... and N != 0:
        Nsuivant = 0
        for i in range(N):
            Nsuivant ...
        N = ...
        k += 1
    return N

print(simul_N(10,L1))
print(simul_N(10,L2))

""" Remarque : afin de tracer un graphique repésentant 
la proba d'extinction en fonction du nombre de générations, 
on ne va pas utiliser la fonction simul_N
mais la réécrire au sein du programme"""

M = 100#nombre de répétitions de l'expérience
T = 50#durée de la simulation
L = L1 # Loi du nombre de descendants
occurence = np.zeros(T+1)#occurences[k] compte le nombre de fois 
                          # où il n'y a pas eu extinction à l'étape k
                          #au cours des M répétitions
for m in range(M):
    N=1#effectif
    n=0
    while N>0 and n...:
        occurence[n]...
        N_suivant=...
        for i in range(....):
            N_suivant=....
        N=....
        n=...
freq=np.ones(T+1)-occurence/M #on veut la fréquence d'extinction
plt.clf()
plt.plot(...,freq)