import numpy as np
import matplotlib.pyplot as plt
def i(i0, beta, t): return n*i0/(i0-(i0-n)*np.exp(-n*beta*t))
def s(i0, beta, t_array): return n-i(i0, beta, t_array)
#def s(*args): return n-i(*args)
if __name__=='__main__':
#poblacion total fija
n = 100.0
#Parametros modificables
i0 = 1.0
beta = 0.003
t_array = np.arange(0, 50, 1)
i_array = i(i0, beta, t_array)
s_array = s(i0, beta, t_array)
plt.plot(t_array, i_array, label='Infectados')
plt.plot(t_array, s_array, label='Suceptibles')
plt.grid(True)
plt.legend()
plt.show()
fig = plt.figure(figsize=(16,8))
for color, beta in zip(['r','b','k','g'],[0.01, 0.005, 0.004, 0.002]):
t_array = np.arange(0, 50, 1)
i_array = i(i0, beta, t_array)
s_array = s(i0,beta, t_array)
plt.plot(t_array,i_array, c=color, label='infectados (beta={}'.format(beta))
plt.grid(True)
plt.legend(fontsize=25)
plt.xticks(fontsize=25)
plt.yticks(fontsize=25)
plt.show()
¿Quieres ver más aportes, preguntas y respuestas de la comunidad?