如何优雅实现函数根据输入参数变更行为？替代if else判空方案

针对你给出的函数，这里提供几种更简洁优雅的替代方案，解决原代码分支冗余、重复计算的问题：

方案一：分步构建返回值（最简洁通用） #

核心思路是先计算必选的x，再根据b、c的存在性按需计算y、z，最后统一整理返回格式，避免多分支判断：

def foo(a=None, b=None, c=None):
    if a is None:
        raise ValueError("a cannot be None!")
    
    # 必执行的操作，计算x
    x = operations  # 替换为实际业务逻辑
    
    result = [x]
    # 当b不为None时计算y并加入结果
    if b is not None:
        y = operations
        result.append(y)
    # 当c不为None时计算z并加入结果
    if c is not None:
        z = operations
        result.append(z)
    
    # 单个结果返回值，多个结果返回元组
    return result[0] if len(result) == 1 else tuple(result)

这个写法把重复的x计算提取出来，逻辑线性清晰，新增参数时只需添加对应的判断和计算步骤即可。

方案二：参数组合映射（适合复杂分支扩展） #

把b、c的None组合情况作为键，映射到对应的处理函数，将分支逻辑转为字典查找，结构更直观：

def foo(a=None, b=None, c=None):
    if a is None:
        raise ValueError("a cannot be None!")
    
    # 每个参数组合对应的处理逻辑
    def both_none():
        x = operations
        return x
    
    def only_b():
        x = operations
        y = operations
        return x, y
    
    def only_c():
        x = operations
        z = operations
        return x, z
    
    def both_not_none():
        x = operations
        y = operations
        z = operations
        return x, y, z
    
    # 构建参数组合与处理函数的映射
    handler_map = {
        (False, False): both_none,
        (True, False): only_b,
        (False, True): only_c,
        (True, True): both_not_none
    }
    
    # 根据当前参数的存在性获取对应处理函数
    key = (b is not None, c is not None)
    return handler_map[key]()

这种方式适合分支逻辑差异较大的场景，后续新增参数组合时，只需添加新的键和处理函数，无需修改原有逻辑。

方案三：过滤空值返回（极简写法） #

如果operations的返回值不会是None，可以先计算所有可能的值，再过滤掉空值后返回：

def foo(a=None, b=None, c=None):
    if a is None:
        raise ValueError("a cannot be None!")
    
    x = operations
    y = operations if b is not None else None
    z = operations if c is not None else None
    
    # 过滤掉None值，整理返回格式
    result = [val for val in [x, y, z] if val is not None]
    return result[0] if len(result) == 1 else tuple(result)

这个写法代码量最少，但要确保业务逻辑中operations不会返回None，否则会误过滤有效结果。

内容的提问来源于stack exchange，提问作者timothy.s.lau