区分大小写的返回声明

时间:2015-10-01 03:37:08

标签: python

我对用户输入和return语句有疑问。

例如,如果我希望用户输入他们要输入的颜色,Blueblue

我知道python是区分大小写的,所以大写的Blue和小写的blue不是一回事。

在我制作的程序中,程序随机生成四种不同的颜色。

问题在于,当我将颜色值分配给随机数时,字符串文本颜色是大写的。

是否可以使用return语句接受大写单词和小写单词?

def randomNum():
    num = random.randint(1,7)
    return num

def randomColor():
    if randomNum() == 1:
        return "Blue"
    elif randomNum() == 2:
        return "Green"
    elif randomNum() == 3:
        return "Pink"
    elif randomNum() == 4:
        return "Purple"
    elif randomNum() == 5:
        return "Red"
    else:
        return "Yellow"

1 个答案:

答案 0 :(得分:2)

您必须确保在调用该函数时,以不区分大小写的方式进行比较。比较字符串时尝试这个:

randomColor().lower() == expected_color.lower()

此外,您只需返回小写字符串:

return "blue" # same with the others

...并相应地进行比较:

randomColor() == expected_color.lower()