计算列表中乌龟的步数 - python

Count number of steps of turtle in a list - python

我现在已经学习了三个星期 python,但我被困住了。 这是我的代码:(之后是我的问题)

from turtle import Screen, Turtle
from random import randint, choice

def person_characteristics(people):
    """ 
    Gives the turtle 'person' all it's characteristics / values. 
    """
    for person in people:
        person.shape('circle')
        person.shapesize(0.2)
        person.speed('fastest')
        person.penup()
        x = randint(-200, 200) #turtle gets a random position
        y = randint(-200, 200)
        person.setpos(x, y)
        person.showturtle()

def population(population_size):
    """
    Makes a population, by making a list of turtles (persons). 
    population_size = type(int)
    """
    people = []

    for _ in range(population_size):
        people.append(Turtle(visible=False))

    return people

def random_walk(person, step_size, area_size):
    """
    Makes the person walk around randomly within the borders. 
    step_size = type(int) -> determines how big of a step each person takes. 
    area_size = type(int) -> determines how big the area is where the persons are in.
    """
    if -area_size < person.xcor() < area_size and -area_size < person.ycor() < area_size: #if person is within the borders then it moves randomly
        person.right(randint(0, 360))
        person.forward(step_size)
    else:
        person.right(180) #if person is outside the borders it turns around
        person.forward(step_size)

def infect_random(people): 
    """
   Random person gets infected (a red color)
    people = a list of persons achieved from de function population()
    """
    infected = choice(people)
    infected.color('red')
    return infected

screen = Screen()

people = population(100)
person_characteristics(people)

infected_people = []
initial_infected = infect_random(people)
infected_people.append(initial_infected)


counted_infections = 1
#count_steps = 0
#healed_people = []

for _ in range(10): # determines the number of steps = time
    for person in people:
        random_walk(person, 30, 400)

        for infected_person in infected_people:
            if person.pencolor() != 'red' and person.distance(infected_person) < 30:  #if a person gets close to the initial infected person it also
                person.color('red')                                                   #gets infected & added to the list of infected persons
                infected_people.append(person)
                #count_steps +=1 
                #if count_steps = 5:
                    #infected_person.color('green')
                    #healed_people.append(infected_person)
                    #infected_people.remove(infected_person) 
                break 


count_susceptible = len(people) - len(infected_people)                          #counts number of susceptible people
count_infected = len(infected_people)                                           #counts number of infected people

print(count_susceptible)
print(count_infected)

screen.exitonclick()

我想将 infected_person 变为绿色(=已治愈)并在海龟处进入列表 healed_people(并从列表 infected_people 中删除)后执行 5 个步骤。我的想法是使用 if 语句来执行此操作,但这不起作用。我的想法在上面的代码中。我知道它为什么不起作用:现在它计算每个 infected_person 的总步数,而不是单独计算。 我认为可能有一个非常简单的解决方案,但我对 python 还很陌生,所以我不知道该怎么做。谁能帮忙?

提前致谢!

(我不想使用 Class,因为我还没有学会 :)

from turtle import Screen, Turtle
from random import randint, choice

def person_characteristics(people):
    """ 
    Gives the turtle 'person' all it's characteristics / values. 
    """
    for person in people:
        person.shape('circle')
        person.shapesize(0.2)
        person.speed('fastest')
        person.penup()
        x = randint(-200, 200) #turtle gets a random position
        y = randint(-200, 200)
        person.setpos(x, y)
        person.showturtle()

def population(population_size):
    """
    Makes a population, by making a list of turtles (persons). 
    population_size = type(int)
    """
    people = []

    for _ in range(population_size):
        people.append(Turtle(visible=False))

    return people

def random_walk(person, step_size, area_size):
    """
    Makes the person walk around randomly within the borders. 
    step_size = type(int) -> determines how big of a step each person takes. 
    area_size = type(int) -> determines how big the area is where the persons are in.
    """
    if -area_size < person.xcor() < area_size and -area_size < person.ycor() < area_size: #if person is within the borders then it moves randomly
        person.right(randint(0, 360))
        person.forward(step_size)
    else:
        person.right(180) #if person is outside the borders it turns around
        person.forward(step_size)

def infect_random(people): 
    """
   Random person gets infected (a red color)
    people = a list of persons achieved from de function population()
    """
    infected = choice(people)
    infected.color('red')
    return infected

screen = Screen()

people = population(100)
person_characteristics(people)

infected_people = []
people_steps=[0 for _ in range (len(people))]
initial_infected = infect_random(people)
infected_people.append(initial_infected)


counted_infections = 1

for _ in range(10): # determines the number of steps = time
    for person in people:
        random_walk(person, 30, 400)
        people_steps[people.index(person)]+=1
        if people_steps[people.index(person)]==5 and person.pencolor()=='red':
            person.color('green')
            infected_people.remove(person) 
        for infected_person in infected_people:
            if person.pencolor() != 'red' and person.distance(infected_person) < 30:  #if a person gets close to the initial infected person it also
                person.color('red')                                                   #gets infected & added to the list of infected persons
                infected_people.append(person)
                people_steps[people.index(person)]=0
                break 


count_susceptible = len(people) - len(infected_people)                          #counts number of susceptible people
count_infected = len(infected_people)                                           #counts number of infected people

print(count_susceptible)
print(count_infected)

screen.exitonclick()

您可以使用辅助数组来保存人的步数。如果以相同的大小创建它,则可以使用索引。例如:

人=[person1, person2, person3]

people_steps=[number_of_steps1, number_of_steps2, number_of_steps3]

这只是一个图形表示。

但是最好用 class 来做,这样步数作为属性的一部分。

希望对您有所帮助。如果您有任何建议或问题,请告诉我。