Python中如何使用map和lambda优化多条件分支的重复return语句

错误原因说明 #

• 你尝试的return new_data = 'state' + 'not started'写法不符合Python语法，Python 3.7版本不支持海象运算符:=，无法在return语句中同时完成变量赋值操作，必须先赋值再返回。
• 原代码中if not 'null'存在逻辑错误：非空字符串的布尔值为True，not 'null'永远为False，对应分支永远不会执行，建议根据实际需求修改为if f != 'null'或if f is not None这类有效判断。
• 原代码中循环遍历transformers['fields']但仅判断第一个元素的取值，循环无实际作用，优化时可以直接取第一个元素处理。

优化方案 #

方案1：字典映射（更推荐，可读性高、易维护） #

这种方案比map+lambda更适合固定key匹配的场景，新增分支只需要在字典中新增对应项即可，无需重复编写if/elif逻辑：

def querying(collection, transformers):
    new_data = ''
    # 定义状态映射规则，需要动态计算的值用lambda包装
    state_map = {
        'open': lambda: 'state' + 'open',
        'closed': lambda: 'state' + 'closed',
        'mid': lambda: 'state' + 'mid',
        'not started': lambda: 'state' + output_of_another_function()
    }
    if transformers and 'fields' in transformers:
        first_field = transformers['fields'][0]
        # 此处将原错误的`if not 'null'`替换为合理判断，可根据实际业务调整
        if first_field != 'null':
            if first_field in state_map:
                new_data = state_map[first_field]()
                return new_data
        else:
            return

方案2：map+lambda实现（满足指定技术要求） #

如果必须使用map和lambda实现，可参考如下写法：

def querying(collection, transformers):
    new_data = ''
    # 定义匹配规则和对应的值生成函数
    match_rules = [
        ('open', lambda: 'state' + 'open'),
        ('closed', lambda: 'state' + 'closed'),
        ('mid', lambda: 'state' + 'mid'),
        ('not started', lambda: 'state' + output_of_another_function())
    ]
    if transformers and 'fields' in transformers:
        first_field = transformers['fields'][0]
        # 此处将原错误的`if not 'null'`替换为合理判断，可根据实际业务调整
        if first_field != 'null':
            # 用map遍历所有匹配规则，生成匹配结果列表
            match_result = list(map(lambda rule: rule[1]() if rule[0] == first_field else None, match_rules))
            # 过滤无效值，取第一个匹配成功的结果
            valid_result = [item for item in match_result if item is not None]
            if valid_result:
                new_data = valid_result[0]
                return new_data
        else:
            return

注意事项 #

两种方案都满足「所有分支先赋值给new_data变量再返回」的要求，你可以根据实际场景选择。如果后续需要新增条件分支，只需要在映射规则中新增对应项即可，不需要重复写if/elif判断逻辑。

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