count = 0
for count in range(turns):
my_world.update_world(count)
if count % frequency == 0:
#plt.bar(['Infected', 'Not'], [int(my_world.get_number_infected()), int(my_world.get_number_well())])
print(count)
print("{}. Number infected: {}".format(
count, my_world.get_number_infected()))
print("{}. Number well: {}".format(count,
my_world.get_number_well()))
if count % train_new_doctor_frequency == 0:
my_world.train_doctor()
infected, healthy, doctors = my_world.create_coord_list()
my_world.plot(infected[2:], healthy[2:], doctors, count)
plt.show()
infectedCount.append(my_world.get_number_infected())
healthyCount.append(my_world.get_number_well())
if my_world.get_number_well() == 0:
print('Finished...')
break
count += 1
print(healthyCount)
plt.plot(range(count), infectedCount)
plt.show()
