Python Input Statement

Mastering User Input in Python

User input is a crucial component of many Python programs, enabling interaction and customization. In this comprehensive guide, we’ll explore the various aspects of handling user input in Python, covering the basics, type conversion, input validation, and practical examples.

Input Function Definition

The input() function is used to receive input from the user. It displays a prompt and waits for the user to enter data. The entered data is then returned as a string.

Diagram:

diagram how input() works mermaid

  graph LR
    A[Show prompt] --> B[Wait for user to type + press Enter]
    B --> C[Return the text as a str]
    C --> D{Need a number?}
    D -->|Yes| E[Convert with int / float]
    D -->|No| F[Use the string directly]

The flow of reading user input

Caution

input() always returns a str, even if the user types digits. input() plus arithmetic without converting first leads to bugs: "5" + "3" is "53", not 8. Wrap it in int() or float() when you need a number.

The syntax of the input() function is as follows:

input.py

input([prompt])

There is one optional parameter, prompt, which is the string displayed to the user. If no prompt is provided, the function simply waits for the user to enter data.

::: tip input() function returns a string. If you expect a different data type, such as an integer or a float, you need to perform type conversion.

for example:

input.py

name = input("Enter your name: ")
age = int(input("Enter your age: "))
height = float(input("Enter your height in meters: "))
 
print(type(name))
print(type(age))
print(type(height))

Output:

command

C:\Users\Your Name> python input.py
Enter your name: John
Enter your age: 25
Enter your height in meters: 1.75
<class 'str'>
<class 'int'>
<class 'float'>

Basics of User Input

In Python, the input() function is the gateway to capturing user input. It prompts the user to enter data, and whatever is entered is treated as a string. Here’s a simple example:

input.py

user_name = input("Enter your name: ")
print("Hello, " + user_name + "!")

Output:

command

C:\Users\Your Name> python input.py
Enter your name: John
Hello, John!

In this snippet, the input() function displays the prompt (“Enter your name: ”) and waits for the user to input their name. The entered value is then stored in the variable user_name and used in the subsequent print() statement.

Note

raw_input() function is used in Python 2.x to receive user input. In Python 3.x, raw_input() was renamed to input().

input.py

user_name = raw_input("Enter your name: ")
print("Hello, " + user_name + "!")

Output:

command

C:\Users\Your Name> python input.py
Enter your name: John
Hello, John!

Type Conversion of User Input

By default, the input from the user is treated as a string. If you expect a different data type, such as an integer or a float, you need to perform type conversion. Here’s an example:

input.py

age = int(input("Enter your age: "))
height = float(input("Enter your height in meters: "))
 
print(type(age))
print(type(height))

Output:

command

C:\Users\Your Name> python input.py
Enter your age: 25
Enter your height in meters: 1.75
<class 'int'>
<class 'float'>

In this example, int() and float() functions are used to convert the user input to integer and float data types, respectively.

Reading Multiple Values at Once

To read several values from a single line, use str.split() to break the input on whitespace (or any separator). Combine it with map() to convert them all in one step.

multi_input.py

# User types: 3 7 1 9
numbers = input("Enter numbers separated by spaces: ").split()
print(numbers)                       # ['3', '7', '1', '9']  (strings)
 
# Convert every item to int
numbers = list(map(int, input("Enter numbers: ").split()))
print(sum(numbers))                  # 20

Unpacking works too, when you know exactly how many values to expect:

unpack_input.py

x, y = input("Enter two values: ").split(",")   # split on a comma
print(x, y)

Tip

User input often has stray spaces. Clean it with .strip() before using it: name = input("Name: ").strip().

Handling User Input Validation

Ensuring that the user provides valid input is crucial for the robustness of your program. Using conditional statements and exception handling, you can validate and handle potential errors gracefully:

input.py

while True:
    try:
        age = int(input("Enter your age: "))
        if age < 0:
            print("Please enter a non-negative age.")
        else:
            break  # Exit the loop if the input is valid
    except ValueError:
        print("Invalid input. Please enter a valid integer.")

Output:

command

C:\Users\Your Name> python input.py
Enter your age: -5
Please enter a non-negative age.
Enter your age: 25

In this example, a while loop is used to continually prompt the user for input until a valid integer is entered. The try and except blocks handle potential errors, such as non-integer inputs.

Input in Script Execution

Python scripts can also receive input through command-line arguments using sys.argv. This allows users to provide input when executing a script:

input.py

import sys
 
if len(sys.argv) > 1:
    user_name = sys.argv[1]
    print("Hello, " + user_name + "!")
else:
    print("Please provide your name as a command-line argument.")

Output:

command

$ python input.py John
Hello, John!

Or in Windows:

command

C:\Users\Your Name> python input.py John
Hello, John!

In this script, if additional arguments are provided during execution, the first argument is used as the user’s name. Otherwise, a prompt is displayed.

Security Considerations

When dealing with user input, it’s crucial to handle it securely to prevent issues like code injection or unintended consequences. If user input is used in database queries or other sensitive operations, consider using parameterized queries or input validation techniques to enhance security.

Conclusion

Handling user input in Python opens up a world of possibilities for creating dynamic and interactive programs. By utilizing the input() function, performing type conversion, validating input, and considering security measures, you can create robust and user-friendly applications. Whether you’re building command-line tools or interactive scripts, understanding how to effectively interact with users enhances the overall user experience.

---

Try it: Input Exercises

Note

DataCamp Light does not support interactive input() prompts — instead, we simulate user input by pre-assigning values so you can practice type conversion and logic.

Exercise 1 – Simulate Name Input

pch.exerciseTag pch.exerciseTitle

Exercise 2 – Simulate Integer Input

pch.exerciseTag pch.exerciseTitle

Exercise 3 – Simulate Multiple Inputs

pch.exerciseTag pch.exerciseTitle

As you continue your Python journey, explore more advanced user input techniques, integrate them into your projects, and ensure a smooth and error-resistant user interaction. For additional guidance and hands-on examples, check out our tutorials on Python Central Hub!