如何检查输入是否等于某个整数或整数？

Question

我知道如何检查输入是否为整数但不是特定的整数。例如，对于我的代码，我想检查输入是否等于1,2或3，然后如果输入不等于1,2或3，则要求用户再次输入。 / p>

非常感谢帮助：）

Answer 1

那将是：

if var in (1, 2, 3):

或任何整数：

if isinstance(var, int):

Answer 2

def restricted_input(prompt, values):
    assert values
    while True:
        typed = input(prompt)
        for acceptable in values:
            if typed == str(acceptable):
                return acceptable
        else:
            print("Acceptable values are from ", values)

然后在你的程序中......

choice = restricted_input("Enter a number: ", {1, 2, 3})