Simulation pédagogique de l'évolution de la pandémie

CoronaSimulation.py

import math
import matplotlib.pyplot as plt

X  = 50
X0 = 30
L  = 30000
K  = 0.5

#X  indique le temps, supposons qu'il commence au jour 0 et s'arrête après 50 jours.
#X0 indique le moment où la croissance exponentielle s'arrête, disons après 30 jours.
#L  indique la valeur maximale pouvant être par la courbe, disons 4.5 10^-4 de la population française, soit 30000
#K  indique la pente de la courbe : disons 0.5

##Functions

def f(L, X, X0, K):
    return (L / (1+math.exp(-K * (X-X0))))

def coronaEvolution(L, X0, K, X):
    res = []
    for t in range(X) :
        res.append(f(L, t, X0, K))
    return res

##Programm Run :

####One Simulation :

sim1 = coronaEvolution(L, X0, K, X)

plt.plot(sim1)
plt.ylabel('Nombre de personnes infectées')
plt.xlabel('Jours')
plt.show()

####More Simulations :

nbSimulations = 5
plt.ylabel('Nombre de personnes infectées')
plt.xlabel('Jours')

for i in range(nbSimulations):
    plt.plot(coronaEvolution(L, X0*(1+(i/nbSimulations)), K+(i/nbSimulations), X), label="Scénario=%d"%(i+1,))

leg = plt.legend(loc='upper left', ncol=1, mode=None, shadow=False, fancybox=False)
leg.get_frame().set_alpha(0.9)