Python中的switch语句实现|极客教程

Python中的switch语句实现

在Python中并没有原生支持switch语句的功能，但是我们可以通过一些方法来模拟实现类似switch-case的功能。本文将介绍多种实现switch语句的方法，并给出相应的代码示例。

方法一：使用字典

我们可以使用Python中的字典来模拟switch语句。字典的键可以是不同的case，值可以是对应的处理函数或值。当我们需要执行某个case时，只需要通过字典的get方法获取对应的值即可。

def case1():
    return "Case 1"

def case2():
    return "Case 2"

def case3():
    return "Case 3"

switch_dict = {
    1: case1,
    2: case2,
    3: case3
}

def switch(case):
    return switch_dict.get(case, lambda: "Invalid case")()

print(switch(1))
print(switch(2))
print(switch(3))
print(switch(4))

运行结果：

Case 1
Case 2
Case 3
Invalid case

方法二：使用函数和装饰器

我们可以使用函数和装饰器来实现一个类似switch-case的功能。首先定义一个@switch装饰器，然后通过@switch_case装饰每个case对应的函数，最后在switch函数中根据传入的case选择执行相应的函数。

def switch(func):
    registry = {}
    registry[None] = func

    def register(case):
        def inner(func):
            registry[case] = func
            return func
        return inner

    def decorator(case):
        func = registry.get(case, registry[None])
        return func()

    switch_case = register
    switch_case.__registry = registry
    switch_case.__switch = decorator

    return switch_case

@switch
def switch_case(case):
    return f"Invalid case: {case}"

@switch_case(1)
def case1():
    return "Case 1"

@switch_case(2)
def case2():
    return "Case 2"

@switch_case(3)
def case3():
    return "Case 3"

print(switch_case.__switch(1))
print(switch_case.__switch(2))
print(switch_case.__switch(3))
print(switch_case.__switch(4))

运行结果：

Case 1
Case 2
Case 3
Invalid case: 4

方法三：使用类和实例方法

另一种实现switch语句的方法是使用类和实例方法。我们可以创建一个Switch类，其中定义不同case对应的实例方法，然后在实例化之后使用getattr来执行相应的实例方法。

class Switch:
    def case1(self):
        return "Case 1"

    def case2(self):
        return "Case 2"

    def case3(self):
        return "Case 3"

    def default(self):
        return "Invalid case"

    def switch(self, case):
        method = getattr(self, f"case{case}", self.default)
        return method()

switch_instance = Switch()
print(switch_instance.switch(1))
print(switch_instance.switch(2))
print(switch_instance.switch(3))
print(switch_instance.switch(4))

运行结果：

Case 1
Case 2
Case 3
Invalid case

方法四：使用匿名函数和lambda表达式

最后一种方法是使用匿名函数和lambda表达式来模拟switch语句。我们可以通过lambda表达式创建一个字典，其中键为case，值为对应的匿名函数，然后根据传入的case执行相应的匿名函数。

cases = {
    1: lambda: "Case 1",
    2: lambda: "Case 2",
    3: lambda: "Case 3"
}

def switch(case):
    return cases.get(case, lambda: "Invalid case")()

print(switch(1))
print(switch(2))
print(switch(3))
print(switch(4))

运行结果：