Ограничение области целочисленных значений, принимаемых в качестве входных данных в Python

Я создаю игру из командной строки, используя python. Основная особенность этой игры - получить вводимые пользователем значения 1 или 2 в виде целых значений. Любой другой символ должен быть отклонен. Я использовал try-except & if-else condition чтобы сделать это, как показано ниже. Я хочу знать, есть ли лучший способ сделать это одной строкой или каким-либо другим способом, не создавая отступа для всей кучи кода.

if __name__ == '__main__':
# INITIALIZE THE TOTAL STICKS , DEPTH OF THE TREE AND THE STARTINGG PLAYER
i_stickTotal = 11 # TOTAL NO OF STICKS IN THIS GAME
i_depth = 5 # THE DEPTH OF THE GOAL TREEE THE COMPUTER WILL BUILD
i_curPlayer = 1 # THIS WILL BE +1 FOR THE HUMAN AND -1 FOR THE COMPUTER
print("""There are 11 sticks in total.\nYou can choose 1 or 2 sticks in each turn.\n\tGood Luck!!""")
# GAME LOOP
while i_stickTotal > 0:
    print("\n{} sticks remain. How many would you pick?".format(i_stickTotal))
    try:
        i_choice = int(input("\n1 or 2: "))
        if  i_choice - 1 == 0 or i_choice - 2 == 0:            
            i_stickTotal -= int(i_choice)
            if WinCheck(i_stickTotal, i_curPlayer):
                i_curPlayer *= -1
                node = Node(i_depth, i_curPlayer, i_stickTotal)
                bestChoice = -100
                i_bestValue = -i_curPlayer * maxsize

                #   Determine No of Sticks to Remove

                for i in range(len(node.children)):
                    n_child = node.children[i]
                    #print("heres what it look like ", n_child.i_depth, "and",i_depth)
                    i_val = MinMax(n_child, i_depth-1, i_curPlayer)
                    if abs(i_curPlayer * maxsize - i_val) <= abs(i_curPlayer*maxsize-i_bestValue):
                        i_bestValue = i_val
                        bestChoice = i
                        #print("Best value was changed @ ", i_depth, " by " , -i_curPlayer, " branch ", i, " to ", i_bestValue)



                bestChoice += 1
                print("Computer chooses: " + str(bestChoice) + "\tbased on value: " + str(i_bestValue))
                i_stickTotal -= bestChoice
                WinCheck(i_stickTotal, i_curPlayer)
                i_curPlayer *= -1
            else:
                print("You can take only a maximum of two sticks.")

    except:
        print("Invalid input.Only Numeric Values are accepted")
Источник

2 ответа

Решение

Вы можете создать функцию для проверки ввода пользователя и использовать приведенный ниже код.

while True:
    var = int(input('Enter value (1 or 2) - '))
    if var not in range(1, 3):
        print('Invalid entry, please try again...')
        continue
    else:
        break
Источник

Напишите функцию, которая зацикливается, вызывая input, пока значение не будет соответствовать вашему ограничению. Возможно назовите этоget_user_input. Затем вызовите это в своей основной функции вместоinput. Для дополнительной ценности передайте лямбда-выражение в эту функцию в качестве предиката для проверки вводимого пользователем значения - это сделаетget_user_input более общий.

Источник
Другие вопросы по тегам