How to Limit User Input to Only Integers in Python for a Multiple Choice Survey

Introduction

In Python programming, especially when creating interactive applications like surveys, it's common to require user input to be restricted to specific types, such as integers. This ensures data integrity and prevents errors. This article explores two effective methods to limit user input to only integers, with a focus on a multiple-choice survey scenario.

Method 1: Direct Try-Except Loop

The most straightforward approach is to use a try-except block within a loop. This method continuously prompts the user until a valid integer is entered.

def survey():
    print('1) Blue')
    print('2) Red')
    print('3) Yellow')
    
    while True:
        try:
            question = int(input('Out of these options (1,2,3), which is your favourite? '))
            break
        except ValueError:
            print("That's not a valid option!")
    
    if question == 1:
        print('Nice!')
    elif question == 2:
        print('Cool')
    elif question == 3:
        print('Awesome!')
    else:
        print('That\'s not an option!')

In this code, the int() function attempts to convert the input string to an integer. If successful, the loop breaks; if a ValueError is raised (e.g., due to non-numeric input), the except block catches it and prints an error message, prompting the user again.

Method 2: Generic Helper Function

For scenarios where multiple inputs need validation, a reusable helper function can be implemented. This approach abstracts the validation logic, making the code cleaner and more maintainable.

def _input(message, input_type=str):
    while True:
        try:
            return input_type(input(message))
        except ValueError:
            pass

if __name__ == '__main__':
    _input("Only accepting integer: ", int)
    _input("Only accepting float: ", float)
    _input("Accepting anything as string: ")

This function takes a message and an input type (defaulting to string) and repeatedly prompts the user until valid input is provided. It can be easily integrated into various parts of a program.

Comparison and Best Practices

Method 1 is ideal for simple, one-off validation, while Method 2 offers reusability and scalability. Best practices include specifying the exact exception (e.g., ValueError) to avoid catching unintended errors and providing clear error messages to users.

Conclusion

Limiting user input to integers in Python can be efficiently achieved through try-except mechanisms. By employing loops and error handling, developers can create robust interactive applications that handle invalid inputs gracefully.