如何在 Python 中指定函数类型？

如何为类型提示指定函数类型

简介

类型提示是 Python 中的一项强大功能，它允许你向代码添加元数据，以帮助捕捉潜在错误并提高代码的可读性。对于函数类型，你可以使用 Callable 类型注释来指定函数的参数类型和返回值类型。

使用 Callable 类型注释

Callable 类型注释的语法如下：

Callable[[arg1_type, arg2_type, ...], return_type]

其中：

• arg1_type, arg2_type, ... 是函数参数的类型注释。
• return_type 是函数返回值的类型注释。

例如，以下代码使用 Callable 类型注释指定了一个函数类型，该函数接受两个整数作为参数并返回一个整数：

from typing import Callable

def add_numbers(a: int, b: int) -> int:
    """Add two numbers together."""
    return a + b

# 使用 Callable 类型注释指定函数类型
my_function: Callable[[int, int], int] = add_numbers

使用类型别名

为了简化函数类型注释，你可以使用类型别名。例如，以下别名定义了一个 IntToInt 函数类型，它接受一个整数作为参数并返回一个整数：

from typing import TypeAlias

IntToInt: TypeAlias = Callable[[int], int]

使用类型别名，你可以像下面这样指定函数类型：

my_function: IntToInt = add_numbers

注意事项

• Callable 类型注释只能指定函数的参数类型和返回值类型，不能指定函数的名称或文档字符串。
• Callable 类型注释不能用于指定可变长度的参数列表或参数。
• Callable 类型注释不能用于指定带有默认值的函数参数。

结论

使用 Callable 类型注释可以指定函数类型，从而帮助你捕捉潜在错误并提高代码的可读性。通过使用类型别名，你可以进一步简化函数类型注释。

常见问题解答

1. 什么是类型提示？
   类型提示是一种向 Python 代码添加元数据以帮助捕捉错误的技术。
2. 如何指定函数类型？
   使用 Callable 类型注释或类型别名指定函数类型。
3. Callable 类型注释的语法是什么？

   Callable[[arg1_type, arg2_type, ...], return_type]
   

4. 是否可以使用类型别名来简化函数类型注释？
   是的，你可以使用类型别名来简化函数类型注释。
5. Callable 类型注释有什么限制？
   Callable 类型注释不能指定函数名称、文档字符串、可变长度参数列表、关键字参数或带有默认值的函数参数。