Python函数优化：如何优雅传递参数并去除条件判断语句？

首先纠正原代码里的一个错误：func2的elif x == 'b'分支里调用func2(b=2)会触发无限递归，应该改为调用func1(b=2)。

接下来提供两种可行的实现方式，满足你无需传递字符串参数、移除if语句的需求：

方法一：直接用关键字参数传递 #

通过让func2接受关键字参数并透传给func1，彻底去掉分支判断：

def func1(a = 0, b = 0):
    return a + b**2

def func2(**kwargs):
    # 可选：添加参数校验，确保只传入a或b其中一个
    if len(kwargs) != 1 or not (('a' in kwargs) ^ ('b' in kwargs)):
        raise ValueError("只能传入a或b其中一个参数")
    return func1(**kwargs)

# 调用示例
print(func2(a = 2))  # 输出 2
print(func2(b = 2))  # 输出 4

这种方式直接通过关键字参数明确指定要传递的参数，完全不需要字符串判断。

方法二：使用偏函数预先绑定参数 #

如果不需要func2作为中间层，可以用functools.partial直接创建绑定好参数的函数，更简洁：

from functools import partial

def func1(a = 0, b = 0):
    return a + b**2

# 预先绑定参数，生成对应功能的函数
func_a = partial(func1, a=2)
func_b = partial(func1, b=2)

# 调用示例
print(func_a())  # 输出 2
print(func_b())  # 输出 4

这种方式直接跳过了func2的中间逻辑，把不同参数绑定的逻辑提前固化成独立函数，调用时更直观。

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