python - 编写 Long "if"语句的更好方法？

Question

你好，这可能是一个非常基本的问题，但因为我是编程和 python 的新手，而且我真的很想学习，所以我问。

我正在制作一个程序，从用户那里获取他所拥有的“扑克牌”套装的输入。程序只接受正确的花色。

例如；

Diamonds, Hearts, Clubs, Spades

如果用户输入任何其他内容，例如“三角形”，程序将返回“错误输入”。

这是我目前得到的:

if suit == "Diamonds":
    return "Accepted"
if suit == "Hearts":
    return "Accepted"
if suit == "Clubs":
    return "Accepted"
if suit == "Spades":
    return "Accepted"
else:
    return "Wrong input"

我的问题是，有没有比经历为每个 Suit 制作一个全新的“if”语句这一繁琐的过程更好的写法。

Answer 1

您可以使用 in 来检查 suit 是否在列表、元组或一组已接受的花色中:

if suit in {"Diamonds", "Hearts", "Clubs", "Spades"}:
    return "Accepted"
else:
    return "Wrong input"

您还可以使用元组(与(...))、list([...]) 或 frozenset 而不是 set ({...})。