如何实现根据函数参数数量返回不同计算结果的功能

Hey there! Let's fix that parameter counting issue first—it's simpler than you think!

When you use *args in a function definition, args is actually a tuple that holds all the positional arguments passed in. So to get the number of arguments, you just need to use the built-in len() function on it.

Here's how to adjust your code:

def function(*args):
    num_of_args = len(args)  # This is the correct way to count arguments
    if num_of_args == 1:
        result = args[0]  # Access the first (and only) argument
    elif num_of_args == 2:
        a, b = args  # Unpack the tuple into named variables for readability
        result = a + b
    elif num_of_args == 3:
        a, b, c = args
        result = a + b + c  # Replace with your actual calculation logic
    elif num_of_args == 4:
        a, b, c, d = args
        result = a + b + c + d
    elif num_of_args == 5:
        a, b, c, d, e = args
        result = a + b + c + d + e
    else:
        # Optional: Handle invalid argument counts
        raise ValueError("Please pass between 1 and 5 arguments")
    return result

A quick tip: Unpacking args into named variables (like a, b = args) makes your code easier to read than using index access (args[0], args[1]) directly, but both approaches work perfectly well.

Since you mentioned you don't want to focus on optimizing the if statements right now, this setup will get you up and running smoothly. Once you're ready to simplify later, you could explore shortcuts like sum(args) for addition-focused cases, but that's totally optional for your current goal.

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