用Python编写红绿灯

在Python中，我们可以创建一个简单的红绿灯交通信号灯模型，通常会使用类来表示信号灯的不同状态。这里是一个基本的示例：

class TrafficLight:
    def __init__(self):
        self.states = ['red', 'yellow', 'green']
        self.current_state = self.states[0]  # 初始状态设为红灯

    def switch(self):
        if self.current_state == 'red':
            self.current_state = 'yellow'
        elif self.current_state == 'yellow':
            self.current_state = 'green'
        else:  # 如果是绿色，则切换回红色
            self.current_state = 'red'
        print(f'Traffic light is now {self.current_state}.')

# 创建并使用交通灯对象
traffic_light = TrafficLight()
while True:
    traffic_light.switch()  # 模拟不断切换的过程

这个程序创建了一个`TrafficLight`类，有三个状态：红、黄、绿，并在一个无限循环中模拟信号灯的切换。