当字符串包含另一个字符串时提供特定输出的最有效方法

Question

我正在用Python编写一个脚本，它检查一个字符串是否包含一套卡片组，因为没有开关/大小写，而且我真的不知道如何用字典来完成它，我有点被难住了

现在我只有4个if语句：

def colorcheck(inp):
    if "Heart" in inp:
        return "Red"
    if "Diamond" in inp:
        return "Red"
    if "Club" in inp:
        return "Black"
    if "Spade" in inp:
        return "Black"
    return "Error"

它给出了正确的结果，但我觉得这可以改进一点

Answer 1

请尝试以下代码：

colors = {
    'Heart': 'Red',
    'Diamond': 'Red',
    'Club': 'Black',
    'Spade': 'Black',
}


def colorcheck(inp):
    for card, color in colors.items():
        if card in inp:
            return color
    return "Error"


print(colorcheck("Ace of Spades"))

输出：

Black