如何在Python中实现同一输入变量兼容整数与浮点数输入？

Great question! The issue with your original code is that the isdigit() method only recognizes strings made entirely of digits (no decimal points), which is why it rejects float inputs like "3.5". Here's a modified version of your function that handles both integers and floats while keeping the non-negative validation logic intact:

def interest_rate(): 
    while True: 
        rate_input = input("Enter the annual interest rates: ") 
        try:
            # Convert input to float (works for both integer and float strings)
            rate = float(rate_input)
            
            # Check for non-negative value
            if rate >= 0:
                # Optional: Return int if the value is a whole number to match original behavior
                if rate.is_integer():
                    return int(rate)
                return rate
            else:
                print("Please enter an interest rate of 0 or greater")
        except ValueError:
            # Triggered if input can't be converted to a number
            print("Please enter a valid number")

Key Changes Explained: #

• Replaced isdigit() with try-except: This is a more robust approach for handling numeric input. By directly attempting to convert the input string to a float, we automatically support both integers and floats. If the conversion fails (e.g., input is letters or symbols), we catch the ValueError and show an error message.
• Preserved non-negative validation: After successful conversion, we still check if the rate is ≥ 0, just like your original code.
• Optional integer conversion: The rate.is_integer() check lets us return an integer when the input is a whole number (like "5"), matching your original function's behavior, while returning a float for decimal values (like "5.25"). If you don't need this type distinction, you can skip this step and just return rate directly.

This modification ensures your function accepts all valid non-negative numeric inputs, whether they're integers or floats.

内容的提问来源于stack exchange，提问作者Naveen