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

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