如何檢查一個值是否存在於多個列表之一中

Question

我有以下代碼：

answer = int(input("input number in range 1-6: "))
yes = [1, 2, 3]
no = [4, 5, 6]
while answer not in yes:
    print("what?")
    answer = int(input())
else:
    if answer in no:
        print("no")
    elif answer in yes:
        print("yes")

如果數字在yes no中，我希望while循環終止。 如何正確地將第二個列表包含在while循環條件中？

Answer 1

您可以使用加法運算符連接列表：

while answer not in yes + no:

Answer 2

您還可以使用and ：

while answer not in yes and answer not in no:

此外，還有其他方法可以連接yes和no ：

• while answer not in [*yes, *no]:
• 您可以添加import itertools並使用while answer not in itertools.chain(yes, no):
• while answer not in [j for i in zip(yes, no) for j in i]: