Python泛型容器转换函数的类型提示：如何标注参数与返回值？

针对你给出的类型转换函数，我们可以通过typing模块的泛型工具完成准确的类型标注，以下分Python 3和Python 2.7两种情况说明：

核心思路 #

函数接收一个任意元素类型的可迭代对象，以及一个可接受该可迭代对象作为构造参数的容器类，最终返回该容器类的实例。我们用类型变量约束元素类型和容器类型的关联性，保证类型推断的准确性。

Python 3 版本 #

先导入必要的类型工具，再完成函数标注：

from typing import Iterable, TypeVar, Type

# 定义元素类型变量
T = TypeVar('T')
# 定义容器类型变量，绑定到容纳T类型元素的可迭代容器
C = TypeVar('C', bound=Iterable[T])

def foo(itr: Iterable[T], cast_type: Type[C]) -> C:
    return cast_type(itr)

Python 2.7 版本 #

Python 2.7通过注释提供类型提示，写法如下：

from typing import Iterable, TypeVar, Type

T = TypeVar('T')
C = TypeVar('C', bound=Iterable[T])

def foo(itr, cast_type):
    # type: (Iterable[T], Type[C]) -> C
    return cast_type(itr)

标注说明 #

• T：代表可迭代对象中的元素类型，确保输入的可迭代对象和输出的容器实例使用相同的元素类型。
• C：代表目标容器类型，通过bound=Iterable[T]约束它必须是能容纳T类型元素的可迭代容器。
• Type[C]：标注cast_type参数，表示它是C类型的类（而非实例），且该类支持传入Iterable[T]来构造实例。
• 返回值C：明确函数返回的是cast_type构造出的目标容器实例。

示例验证 #

调用示例中：

• foo([1,2], cast_type=set) 会被推断返回 set[int]
• foo(set([1,2]), cast_type=list) 会被推断返回 list[int]

内容的提问来源于stack exchange，提问作者Pontus Hultkrantz