Python中如何正确使用elif/else实现stop_light交通信号灯函数

代码问题 #

现有代码满足变色条件的分支使用了print而非return，仅在时长不足的分支有返回值，不符合函数要求。原代码末尾的else分支已经实现了时长不足时返回当前颜色的逻辑，无需额外修改。

修正后代码 #

def stop_light(color, seconds):
    if color == "green" and seconds > 60:
        return "yellow"
    elif color == "yellow" and seconds > 5:
        return "red"
    elif color == "red" and seconds > 55:
        return "green"
    else:
        return color 

# 测试用例
print(stop_light("yellow", 3))  # 输出 yellow
print(stop_light("green", 62))  # 输出 yellow
print(stop_light("red", 56))    # 输出 green

注意事项 #

函数返回值需要接收才能使用，可直接打印函数调用结果，或者赋值给变量后再做后续处理。

内容的提问来源于stack exchange，提问作者Black