如何在python上降低嵌套if的复杂度

在Python中，降低嵌套if语句的复杂度可以通过多种方法实现，包括使用函数分解、提前返回、使用字典映射、多态等。以下是一些具体的策略和示例代码：

1. 函数分解

将复杂的逻辑分解成多个小函数，每个函数只负责一个简单的任务。

def is_valid_input(input):
    return input > 0

def process_positive_input(input):
    # 处理正数输入的逻辑
    return input * 2

def process_negative_input(input):
    # 处理负数输入的逻辑
    return input / 2

def main(input):
    if not is_valid_input(input):
        return "Invalid input"
    
    if input > 0:
        return process_positive_input(input)
    else:
        return process_negative_input(input)

# 示例调用
result = main(5)
print(result)  # 输出: 10

2. 提前返回

通过在函数中尽早返回，减少嵌套层级。

def main(input):
    if input <= 0:
        return "Invalid input"
    
    if input > 0:
        return input * 2
    else:
        return input / 2

# 示例调用
result = main(5)
print(result)  # 输出: 10

3. 使用字典映射

对于一些基于条件的函数调用，可以使用字典来映射条件和对应的函数。

def process_positive(input):
    return input * 2

def process_negative(input):
    return input / 2

def main(input):
    action = {
        True: process_positive,
        False: process_negative
    }.get(input > 0, lambda x: "Invalid input")
    
    return action(input)

# 示例调用
result = main(5)
print(result)  # 输出: 10

4. 多态

如果条件逻辑涉及到对象的行为，可以考虑使用面向对象编程中的多态。

class InputProcessor:
    def process(self, input):
        raise NotImplementedError

class PositiveInputProcessor(InputProcessor):
    def process(self, input):
        return input * 2

class NegativeInputProcessor(InputProcessor):
    def process(self, input):
        return input / 2

def get_processor(input):
    if input > 0:
        return PositiveInputProcessor()
    elif input < 0:
        return NegativeInputProcessor()
    else:
        return None

def main(input):
    processor = get_processor(input)
    if processor is None:
        return "Invalid input"
    return processor.process(input)

# 示例调用
result = main(5)
print(result)  # 输出: 10