diff --git a/repos/887c5b163f5359a6a13a7858e7f484d4/prueba.py b/repos/887c5b163f5359a6a13a7858e7f484d4/prueba.py new file mode 100644 index 0000000..2357a26 --- /dev/null +++ b/repos/887c5b163f5359a6a13a7858e7f484d4/prueba.py @@ -0,0 +1,17 @@ +NEUTRAL = 0 +INFECTED = 1 + +class ControlModelM2(BaseBehaviour): + NEUTRAL = 0 + INFECTED = 1 + def step(self, now): + if self.state['id'] == NEUTRAL: + self.neutral_behaviour() + elif self.state['id'] == INFECTED: + self.infected_behaviour() + … + def infected_behaviour(self): + neutral_neighbors = self.get_neighboring_agents(state_id=INFECTED) + for neighbor in neutral_neighbors: + if random.random() < self.prob_infect: + neighbor.state['id'] = INFECTED