代码收藏家技术教程 2025-02-11

Python 中的模式匹配：深入了解 match 语句

在 Python 3.10 及更高版本中，match 语句引入了一种强大的模式匹配功能，类似于其他语言中的 switch 语句，但功能更为丰富和灵活。本文将详细介绍 match 语句的语法、用法以及一些使用注意事项。

1. `match` 语句的基本语法

match 语句的基本语法如下：

match expression:  
    case pattern1:  
        # 处理 pattern1 匹配的代码  
    case pattern2:  
        # 处理 pattern2 匹配的代码  
    case _:  
        # 默认处理代码（通配符）

expression：要进行匹配的表达式。

pattern：可以是字面值、变量、序列、字典等多种形式的模式。

_：通配符，匹配任何值，通常用于默认情况。

2. 基本用法示例

以下是一些使用 match 语句的示例：

2.1 基本用法

def http_error(status):  
    match status:  
        case 400:  
            return "Bad request"  
        case 404:  
            return "Not found"  
        case 418:  
            return "I'm a teapot"  
        case _:  
            return "Something's wrong with the internet"

2.2 解包赋值

match 语句可以解包元组或列表：

point = (0, 1)  
match point:  
    case (0, 0):  
        print("Origin")  
    case (0, y):  
        print(f"Y={y}")  
    case (x, 0):  
        print(f"X={x}")  
    case (x, y):  
        print(f"X={x}, Y={y}")

2.3 类的模式匹配

可以匹配类的实例并提取属性：

class Point:  
    def __init__(self, x, y):  
        self.x = x  
        self.y = y  

def where_is(point):  
    match point:  
        case Point(0, 0):  
            print("Origin")  
        case Point(0, y):  
            print(f"Y={y}")  
        case Point(x, 0):  
            print(f"X={x}")  
        case Point(x, y):  
            print(f"X={x}, Y={y}")  
        case _:  
            print("Not a point")

3. 使用注意事项

匹配顺序：match 语句会按照从上到下的顺序进行匹配，找到第一个匹配的模式后执行对应的代码块，后续的 case 将被忽略。

通配符 _：使用 _ 作为通配符可以捕获所有未匹配的情况，通常放在最后。

模式的灵活性：match 语句支持多种模式，包括字面值、序列、字典、类实例等，能够进行复杂的匹配和解包。

守卫子句：可以在 case 后添加 if 条件来进一步限制匹配：

match point:  
    case (x, y) if x == y:  
        print(f"On the diagonal at {x}")  
    case (x, y):  
        print(f"Point at {x}, {y}")

嵌套模式：模式可以嵌套使用，允许更复杂的结构匹配。

类型匹配：match 语句可以直接匹配对象的类型和属性，提供了更强大的功能。

Python 中的模式匹配（match 语句）与其他语言的 switch 语句有几个显著的不同之处：

模式匹配的灵活性：

Python 的 match 语句不仅可以匹配简单的值，还可以匹配复杂的数据结构，如元组、列表和对象属性。这种能力使得 match 更像是模式匹配，而不仅仅是值的比较。

例如，可以使用 match 来解包元组并同时绑定变量，而 switch 语句通常只支持对单一值的比较。

使用 case 和守卫：

在 Python 的 match 语句中，可以使用 case 语句来定义多个匹配模式，并且可以添加守卫条件（if 子句）来进一步限制匹配。这意味着可以在匹配成功后进行额外的条件检查。

例如：

match point:  
    case (x, y) if x == y:  
        print("On the diagonal")  
    case (x, y):  
        print(f"Point at {x}, {y}")

单一匹配与优先级：

match 语句只会执行第一个匹配成功的 case 块，而 switch 语句在没有 break 的情况下会继续执行后续的 case 块，导致“贯穿”行为。

在 Python 中，模式匹配的设计使得代码更清晰，因为每个 case 块是独立的。

通配符和解包：

Python 的 match 语句支持使用通配符（如 _）来匹配任何值，并且可以通过解包来提取值。这使得处理复杂数据结构时更加简洁和直观。

例如：

match data:  
    case {'type': 'error', 'message': msg}:  
        print(f"Error: {msg}")  
    case {'type': 'info', 'message': msg}:  
        print(f"Info: {msg}")