针对python中的列表测试用户输入

Question

我需要测试用户输入是否与列表的元素相同，现在我正在这样做：

cars = ("red", "yellow", "blue")
guess = str(input())

if guess == cars[1] or guess == cars[2]:
        print("success!")

但是我正在使用更大的列表，并且所有这些检查都使if语句增长很多，有没有办法引用多个索引，例如：

if guess == cars[1] or cars[2]

要么

if guess == cars[1,2,3]

阅读列表文档后，我发现不可能引用多个索引，就像我在上面尝试过的那样，当然会发送语法错误。

Answer 1

最简单的方法是：

if guess in cars:
    ...

但是如果您的清单很大，那将会很慢。然后，您应该将汽车列表存储在一组中：

cars_set = set(cars)
....
if guess in cars_set:
    ...

检查某项是否存在集合比检查某项是否存在于列表中要快得多（但这仅在您有很多项并且要进行多次检查时才成为问题）。

（编辑：我假设cars[0]问题中代码的遗漏是偶然的。如果不是，请使用cars[1:]代替cars。）