Как изменить цвет случайной черепахи?

Question

Мне нужно выбрать человека случайным образом из списка, а затем заразить его вирусом (меняя цвет черепахи на красный), когда я нажимаю «х».Затем я должен посчитать, сколько людей заражено.

В настоящее время мой код не меняет цвет любой черепахи, и я также не знаю, как подсчитать, сколько там красной черепахи.

#used to infect 
class Virus:
    def __init__(self, colour, duration):
        self.colour = colour
        self.duration = duration
class Person:
    def __init__(self, world_size):
        self.world_size = world_size
        self.radius = 7
        self.location = turtle.position()
        self.destination = self._get_random_location()
        turtle.penup()
        turtle.setposition(self.location)
        turtle.setheading(turtle.towards(self.destination))
        self.heading = turtle.heading()

    #random locations are used to assign a destination for the person
    #the possible locations should not be closer than 1 radius to the edge of the world 
    def _get_random_location(self):
        x = random.randint( - 349, 349 )
        y = random.randint( - 249, 249 )
        return (x, y)


    #draw a person using a dot.  Use colour if implementing Viruses 
    def draw(self):
        x, y = self.location
        turtle.penup()
        turtle.setposition(x, y - self.radius)
        turtle.pendown()
        turtle.begin_fill()
        self.x = turtle.circle(self.radius)
        turtle.end_fill()

#infect a person with the given virus
    def infect(self, virus):
        v_colour = virus.colour
        turtle.colormode("v_colour")
class World:
    def __init__(self, width, height, n):
        self.size = (width, height)
        self.hours = 0
        self.people = []
        self.infected = 0
        for i in range(n):
            self.add_person()


    #add a person to the list
    def add_person(self):
        person = Person(1)
        self.people.append(person)


    #choose a random person to infect and infect with a Virus when press 'x'
    def infect_person(self):
        random_ppl = random.choice(self.people)
        v = Virus("red", 100)
        random_ppl.infect(v)

    #remove all infections from all people when press'c'
    def cure_all(self):
        turtle.colormode('black')
        self.infected = 0

def draw(self):
        turtle.clear()
        turtle.hideturtle()
        turtle.setheading(0)
        turtle.penup()
        turtle.setposition(-350, -250)
        turtle.pendown()

        for i in range(2):
            turtle.forward(500)
            turtle.right(90)
            turtle.forward(700)
            turtle.right(90)

        for item in self.people:
            item.draw()

        turtle.penup()
        turtle.setposition(-350, 250)
        turtle.write(f'Hours: {self.hours}', move=False, align='left')
        self.count_infected()
        turtle.setposition(0, 250)
        turtle.write(f'Infected: {self.infected}', move=False, align='left')
        turtle.update()



    #Count the number of infected people
    def count_infected(self):
        p = Person(1)
        p.draw()
        dot = p.x
        color = dot.color()
        Color = color[0]
        if Color is 'red':
            self.infected += 1

#code for the keys
class GraphicalWorld:
    """ Handles the user interface for the simulation

    space - starts and stops the simulation
    'z' - resets the application to the initial state
    'x' - infects a random person
    'c' - cures all the people
    """
    def __init__(self):
        self.WIDTH = 800
        self.HEIGHT = 600
        self.TITLE = 'COMPSCI 130 Project One'
        self.MARGIN = 50 #gap around each side
        self.PEOPLE = 200 #number of people in the simulation
        self.framework = AnimationFramework(self.WIDTH, self.HEIGHT, self.TITLE)

        self.framework.add_key_action(self.setup, 'z') 
        self.framework.add_key_action(self.infect, 'x')
        self.framework.add_key_action(self.cure, 'c')
        self.framework.add_key_action(self.toggle_simulation, ' ') 
        self.framework.add_tick_action(self.next_turn)

        self.world = None

    def setup(self):
        """ Reset the simulation to the initial state """
        print('resetting the world')        
        self.framework.stop_simulation()
        self.world = World(self.WIDTH - self.MARGIN * 2, self.HEIGHT - self.MARGIN * 2, self.PEOPLE)
        self.world.draw()

    def infect(self):
        """ Infect a person, and update the drawing """
        print('infecting a person')
        self.world.infect_person()
        self.world.draw()

    def cure(self):
        """ Remove infections from all the people """
        print('cured all people')
        self.world.cure_all()
        self.world.draw()

Сначала домашнее задание вызывает World.infected_people(), поэтому я не знаю, как передать случайным людям Person.infect(), а затем изменить цвет.

Answer 1

У вас есть ряд проблем, но поскольку вы не включили весь свой код, я могу лишь дать вам несколько советов о том, как действовать:

Помните из моего последнего ответа, что одиночная черепаха общая , поэтому установка цвета черепахи на цвет вируса недостаточна. Вам необходимо хранить информацию о вирусе вместе с Person и использовать эту информацию, когда придет время привлечь человека. I.e.:

# to Person.__init__ add the property
self.virus = None

Сделайте вашу Person.infect() функцию просто:

def infect(self, virus):
    self.virus = virus

Обратите внимание, что код, который вы имели: turtle.colormode("v_colour") полностью неверен, поскольку colormode() - неправильный метод и v_colour не должно быть в кавычках. Но нам это все равно не нужно. Теперь в вашем методе Person.draw() добавьте следующее прямо перед turtle.begin_fill():

if self.virus is not None:
    turtle.color(self.virus.colour)
else:
    turtle.color("black")

Когда вы набираете 'x', вы должны начать видеть, как люди становятся красными.

Я также не знаю, как посчитать, сколько там красной черепахи.

Не считайте красных черепах! (Ваш текущий count_infected() совершенно неверен - вы не должны создавать новых людей и ничего рисовать.)

Вы можете перебрать self.people в поисках людей, у которых есть свойство self.virus, которое не None. Но World имеет свойство self.infected, которое вы должны увеличивать при вызове infect_person(). Однако, поскольку вы выбираете человека случайным образом, это не будет работать как есть, поскольку вы можете увеличить счетчик в два раза для одного и того же человека. Вам нужно изменить infect_person(), чтобы продолжить захват случайного человека , пока не найдет того, кто еще не заражен. Имеет ли это смысл, зависит от правил назначения.

Ваша функция World.cure_all() должна быть переписана для циклического перебора self.people, вызывая cure() для каждого человека. Метод Person.cure() должен установить для свойства self.virus значение None.