PYTHON: coches en vertical

myCar1 = 0

myCar2 = 0

def setup():

    size(800, 600)

    global myCar1, myCar2

    myCar1 = Car(color(255, 0, 0), 500, 400, -5)

    myCar2 = Car(color(0, 0, 255), 300, 300, 5)

def draw():

    background(255)

    

    myCar1.drive()

    myCar1.display()

    myCar2.drive()

    myCar2.display()

class Car:

    def __init__(self, c, xpos, ypos, yspeed):

        self.c = c

        self.xpos = xpos

        self.ypos = ypos

        self.yspeed = yspeed

    

    def display(self):

        stroke(0)

        fill(self.c)

        rectMode(CENTER)

        rect(self.xpos, self.ypos, 20, 10)

    

    def drive(self):

        self.ypos = self.ypos + self.yspeed

        if self.ypos > height:

            self.ypos = 0

        if self.ypos < 0:

            self.ypos = height