#Traffic light LEDs using an array

from machine import Pin
import time

# Define GPIO pins for the LEDs
RED = Pin(2, Pin.OUT)
AMBER = Pin(6, Pin.OUT)
GREEN = Pin(10, Pin.OUT)

# Put them in an array
lights = [RED, AMBER, GREEN]

# Helper function to turn all lights off
def all_off():
    for light in lights:
        light.off()

while True:
    # Red
    all_off()
    RED.on()
    time.sleep(3)

    # Red + Amber
    AMBER.on()
    time.sleep(1)

    # Green
    all_off()
    GREEN.on()
    time.sleep(3)

    # Amber
    all_off()
    AMBER.on()
    time.sleep(1)