Python 中使用 TypeVar 实现泛型联合类型推断的正确实践-Python教程-PHP中文网

Python 中使用 TypeVar 实现泛型联合类型推断的正确实践

本文详解为何 mypy 无法从 union[type[a], type[b]] 推导出 homogenous list[a] 或 list[b]，并提供基于 typevar 的泛型解决方案，确保类型安全且语义清晰。

本文详解为何 mypy 无法从 union[type[a], type[b]] 推导出 homogenous list[a] 或 list[b]，并提供基于 typevar 的泛型解决方案，确保类型安全且语义清晰。

在 Python 类型检查（尤其是 mypy）中，一个常见误区是认为 Union[type[A], type[B]] 参数能“动态决定”返回值的具体类型。例如，当函数签名声明返回 Union[list[A], list[B]]，而内部仅调用一次 a() 构造实例时，开发者直觉上期望：若传入 A，则返回 list[A]；若传入 B，则返回 list[B]。但 mypy 的静态类型系统不执行运行时路径分析——它仅基于类型注解做保守推断。

关键问题在于：a: Union[type[A], type[B]] 表示 a 可能是 A 的类对象，也可能是 B 的类对象，但类型检查器无法确定具体是哪一个。因此，表达式 a() 的类型被保守推断为 Union[A, B]（即 A | B），进而导致列表推导式 [a() for x in items] 的结果类型为 list[A | B]，而非所需的 list[A] | list[B]。注意：list[A | B] 与 Union[list[A], list[B]] 在类型系统中完全不等价——前者表示“元素可混杂 A 和 B 的列表”，后者表示“要么是纯 A 列表，要么是纯 B 列表”。

✅ 正确解法：使用约束型 TypeVar（Constrained TypeVar）

通过引入 TypeVar('T', A, B)，我们将函数建模为泛型重载行为，而非单一宽泛联合类型。mypy 将据此为每个具体类型实参生成独立的类型签名：

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

from typing import TypeVar, List, Type

class A: pass
class B: pass

T = TypeVar('T', A, B)

def f(items, t: Type[T]) -> List[T]:
    return [t() for _ in items]

此时，调用行为获得精确类型推断：