def main(): for p1 in p1_range: for p3 in p3_range: # create the simulation object sim = SIRS(p1=p1, p2=0.5, p3=p3, create_anim=False, x_dim=20, y_dim=20) sim.count_infected_sites = True print(p1, p3) inf_array = sim.simulate() avg = np.mean(inf_array) / float(sim.sweep) susc = np.var(inf_array) / float(sim.sweep) with open(out_file, 'a+') as f: f.write(' '.join( (str(p1), str(p3), str(avg), str(susc), '\n')))
def main(): for frac in frac_range: print(frac) # create the simulation object sim = SIRS(p1=0.5, p2=0.5, p3=0.5, create_anim=False, immune_frac=frac, x_dim=20, y_dim=20) sim.count_infected_sites = True inf_array = sim.simulate() #inf_array_sq = [val*val for val in inf_array] avg = np.mean(inf_array) / float(sim.sweep) susc = np.std(inf_array) / float(sim.sweep) with open(out_file, 'a+') as f: f.write(' '.join((str(frac), str(avg), '\n')))