from tinamit.tiempo import EspecTiempo
import numpy as np
import pandas as pd

extern = pd.DataFrame(
    data={'taza de infección': np.arange(0.001, 0.005, (0.005-0.001)/100)},
    index=pd.date_range('2000-01-01', periods=100)
)
res_base = mds.simular(t=EspecTiempo(100, f_inic='2000-01-01'))
res_extern = mds.simular(t=EspecTiempo(100, f_inic='2000-01-01'), extern=extern)

# Visualizar
plt.plot(res_base['Individuos Infectados'].vals, label='Constante')
plt.plot(res_extern['Individuos Infectados'].vals, label='Variable')

plt.legend()
plt.title('Población infectada')
plt.xlabel('Días')
plt.ylabel('Personas')