如何在 Python 中为 Callable 创建一个具有未知数量参数的泛型？

本文介绍了如何使用 typing.TypeVarTuple 和 typing.Unpack 在 Python 中为 Callable 创建一个泛型，以处理未知数量的参数。通过这种方式，我们可以确保函数参数的类型与可迭代对象中元组的类型相匹配，从而实现更精确的类型提示和更健壮的代码。文章提供了一个 starmap 函数的示例，展示了如何应用这些类型提示工具来增强代码的类型安全性。

在 Python 中，当我们需要处理一个接受可变数量参数的 Callable 对象，并且希望确保这些参数的类型与某个可迭代对象中的元组类型相匹配时，可以使用 typing.TypeVarTuple 和 typing.Unpack 来实现。 这两个工具允许我们定义一个泛型，该泛型可以捕获任意数量的类型，并将它们应用于函数参数和可迭代对象中的元组。

下面是一个 starmap 函数的例子，它类似于 itertools.starmap，但提供了更精确的类型提示：

from typing import (
Callable,
Generator,
Iterable,
TypeVar,
TypeVarTuple,
Unpack,
)
T = TypeVar("T")
Ts = TypeVarTuple("Ts")
def starmap(
func: Callable[[Unpack[Ts]], T], iterable: Iterable[tuple[Unpack[Ts]]]
) -> Generator[T, None, None]:
yield from map(func, *zip(*iterable))

在这个例子中：

• T = TypeVar(“T”) 定义了一个类型变量 T，用于表示函数 func 的返回值类型。
• Ts = TypeVarTuple(“Ts”) 定义了一个类型变量元组 Ts，用于表示函数 func 接受的参数类型。
• Callable[[Unpack[Ts]], T] 表示 func 是一个可调用对象，它接受 Ts 中的所有类型作为参数，并返回类型为 T 的值。Unpack[Ts] 用于将 Ts 中的类型解包为单独的参数。
• Iterable[tuple[Unpack[Ts]]] 表示 iterable 是一个可迭代对象，它包含元组，每个元组的类型与 Ts 中的类型相匹配。

示例：

立即学习“Python免费学习笔记（深入）”；

import itertools
def func(x: int, y: int) -> float:
return x + y
args = func, [(1, 2), (3, 4)]
assert list(itertools.starmap(*args)) == list(starmap(*args))

在这个示例中，func 接受两个 int 类型的参数，并返回一个 float 类型的值。iterable 包含两个元组，每个元组都包含两个 int 类型的值。starmap 函数将 func 应用于 iterable 中的每个元组，并将结果生成为一个迭代器。

注意事项：

• typing.TypeVarTuple 和 typing.Unpack 是 Python 3.11 中引入的新特性。如果你使用的是较早的 Python 版本，则无法使用这些特性。
• 在使用 typing.TypeVarTuple 和 typing.Unpack 时，需要确保函数参数的类型与可迭代对象中元组的类型相匹配。如果不匹配，则会导致类型错误。

总结：

通过使用 typing.TypeVarTuple 和 typing.Unpack，我们可以为接受可变数量参数的 Callable 对象创建更精确的类型提示。这可以帮助我们编写更健壮的代码，并减少运行时错误的发生。 这种方法特别适用于需要处理具有不同参数数量的函数的场景， 例如数据处理、函数组合等。