people = [] for i in range(14): for j in range(14): pos = [1.0 / 15 * i + 1.0 / 15, 1.0 / 15 * j + 1.0 / 15] theta = 2 * np.pi * np.random.rand() vel = [np.cos(theta), np.sin(theta)] rad = 0.01 person = Person(pos, vel, radius=rad) people.append(person) people[0].health = 'sick' if quarantine_rate is not None: for ind, person in enumerate(people): if rand() < quarantine_rate and ind != 0: person.quarantined = True sim = Sim([0, 1, 0, 1], people, name=sys.argv[1], transm_prob=transm_prob, death_rate=dr, recovery_time=rt) sim.write_state_to_file('states') skip = 1 while sim.total_sick > 0: sim.proceed(skip) sim.write_state_to_file('states') if print_: print(sim.total_sick) sim.end()