python - 如何在 Python 中正确 elif/else？

Question

我需要编写一个名为 stop_light 的函数来确定 stop_light 是否应该改变颜色，如果是，它应该改变成什么颜色。

• Stoplight 函数应该有两个参数。第一个的值将是“绿色”、“黄色”或“红色”。这表示停车灯的当前颜色。
• 第二个参数告诉函数这种颜色显示了多长时间。如果绿色已显示超过 60 秒，则返回“黄色”。如果黄色已显示超过 5 秒，则返回“红色”。如果红色显示时间超过 55 秒，则返回“绿色”。
• 如果颜色没有显示足够长的时间（例如绿色已经显示了 17 秒），则返回当前颜色。以下是使用不同参数调用函数的一些示例。

我可以做，但我不能做这部分：

• 如果颜色没有显示足够长的时间（例如绿色已显示 17 秒），则返回当前颜色。”

我该怎么做那部分？

这是我的代码

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

stop_light("yellow", 3) 

标签： pythonpython-3.xif-statementboolean

Answer 1

这行得通

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


stop_light("yellow", 3)

或者

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))