Python中能否为解包运算符或可变参数函数标注返回的所有可变参数类型?
可变参数函数的类型标注解决方案
不需要编写大量重载实现,Python 3.11 引入的 PEP 646 可变长度类型元组(TypeVarTuple) 可以完美解决这个场景的类型标注需求,支持任意数量、任意类型的入参推导,也兼容解包传入的场景。
实现代码
# Python 3.11+ 直接从 typing 导入,低版本可以从 typing_extensions 导入兼容 from typing import TypeVarTuple, Unpack Ts = TypeVarTuple('Ts') def test(*args: Unpack[Ts]) -> tuple[*Ts]: return args
效果验证
- 普通传参场景可以正确推导返回值类型:
比如test(1, "a", True)的返回值类型会被识别为tuple[int, str, bool] - 解包场景只要传入的是固定长度的元组类型,也能正确匹配:
注意如果你解包的是# 解包元组字面量可以正常推导 x: tuple[int, int, str] = test(*(1, 2, "4")) # 类型检查通过list类型,由于Python的list默认被推导为可变长度的同构类型(即上述[1,2,"4"]会被识别为list[int | str]),类型检查器会返回tuple[int | str, ...]的推导结果,你可以将待解包对象显式标注为固定长度tuple类型来解决这个问题。
兼容性说明
主流类型检查器(mypy 0.990+、pyright、pytype)均已支持PEP 646语法,需要兼容Python 3.10及以下版本时,安装typing-extensions包导入对应API即可正常使用。
内容的提问来源于stack exchange,提问作者ssice
相关产品推荐
相关产品推荐

