当前位置：首页 > news >正文

【python3】函数注解

news 2025/8/7 3:59:19

Python 函数注解 (Function Annotations)

Python 函数注解 (Function Annotations)
- - 函数注解的基本语法
  - - 基本语法格式
    - 示例
  - 特殊类型注解
  - 注解信息的存储与访问
  - 函数注解的实际用途
  - 注意事项
  - 小结

函数注解是 Python 的一种特性，用于为函数的参数和返回值添加 元数据。注解通常和类型提示（type hinting）相关联，但它们本质上不会影响程序的实际执行，只是作为一种信息记录。

从Python 3.0开始，函数注解就成为了语言的一部分。在Python 3.5中，加入了类型提示（PEP 484），使得函数注解的使用变得更加规范化和流行。

函数注解的基本语法

函数注解的基本形式是使用冒号 : 来为函数的参数指定类型信息，使用 -> 来为函数的返回值指定类型信息。

基本语法格式

def function_name(param_name: annotation, ...) -> return_annotation:# 函数体pass

参数注解：使用 param_name: annotation 的形式为参数进行注解。
返回值注解：使用 -> return_annotation 的形式为返回值进行注解。
注解通常在函数定义中作为可选的元数据存在，不会对实际的代码逻辑产生影响。

示例

def greet(name: str) -> str:return f"Hello, {name}"def add(x: int, y: int) -> int:return x + ydef process_data(data: list[int]) -> dict[str, int]:# 假设这个函数对输入的数据进行处理并返回一个字典result = {str(index): value for index, value in enumerate(data)}return result

greet 函数：
- 参数 name 注解为 str，表示这个参数应该是一个字符串。
- 返回值注解为 str，表示该函数返回一个字符串。
add 函数：
- 参数 x 和 y 注解为 int，表示它们应该是整数。
- 返回值注解为 int，表示返回的结果是一个整数。
process_data 函数：
- 参数 data 注解为 list[int]，表示它是一个包含整数的列表。
- 返回值注解为 dict[str, int]，表示它返回一个键为字符串、值为整数的字典。

特殊类型注解

Python通过 typing 模块提供了一些用于类型注解的工具，以帮助描述更加复杂的数据结构。这些注解工具使得我们可以对更加复杂的类型关系进行描述，例如列表、字典、元组、联合类型等。

以下是一些常见的 typing 模块的类型：

List：用来注解列表，例如 List[int] 表示一个包含整数的列表。
Dict：用来注解字典，例如 Dict[str, int] 表示键是字符串、值是整数的字典。
Tuple：用来注解元组，例如 Tuple[int, str] 表示包含一个整数和一个字符串的元组。
Optional：用来注解可选值，即参数可能为指定类型或 None，例如 Optional[str]。
Union：用来注解参数可以是多种类型之一，例如 Union[int, str] 表示参数可以是整数或字符串。

例如：

from typing import List, Dict, Tuple, Optional, Uniondef get_user_data(user_id: int) -> Optional[Dict[str, Union[str, int]]]:if user_id == 1:return {"name": "Alice", "age": 30}else:return Nonedef get_coordinates() -> Tuple[float, float]:return (37.7749, -122.4194)

get_user_data 函数返回一个 Optional 的字典，这个字典的键是字符串，值是字符串或整数，或者它可能返回 None。
get_coordinates 函数返回一个元组，包含两个浮点数。

注解信息的存储与访问

函数注解信息存储在函数对象的 __annotations__ 属性中。你可以通过访问这个属性来查看注解信息。

例如：

def multiply(a: int, b: int) -> int:return a * bprint(multiply.__annotations__)

输出：

{'a': <class 'int'>, 'b': <class 'int'>, 'return': <class 'int'>}

这里的输出表明 multiply 函数的参数 a 和 b 以及返回值都有类型注解为 int。

函数注解的实际用途

代码文档化：
注解可以使代码更加自文档化，程序员可以一目了然地了解函数期望的输入类型和输出类型，而不必阅读完整的函数实现。
静态类型检查：
使用工具如 mypy，可以通过静态类型检查来发现一些潜在的类型错误。例如，mypy 可以在开发阶段验证函数的使用是否遵循注解要求，有助于提高代码质量。
IDE支持：
现代化的集成开发环境（如 PyCharm, VSCode）可以基于注解提供更好的自动补全、类型推断和错误提示，使得开发者在编写代码时更容易避免错误。