如何编写接收元组并返回各元素可选的同长度元组的函数？

要实现你需要的函数类型标注，Python 3.11+（或通过typing-extensions库兼容旧版本）可以利用TypeVarTuple和Unpack来对元组的每个元素类型进行批量包装：

from typing import TypeVarTuple, Unpack, Optional, Tuple

# 定义元组元素的类型变量元组
Ts = TypeVarTuple('Ts')

def f(tup: Tuple[Unpack[Ts]]) -> Tuple[Unpack[Tuple[Optional[T] for T in Ts]]]:
    # 实际实现：根据自定义条件返回None或原元素，最后转成元组
    return tuple(None if some_condition else el for el in tup)

代码说明 #

• TypeVarTuple('Ts')：用来表示元组中任意数量的元素类型集合
• Tuple[Unpack[Ts]]：匹配输入的任意长度元组，Unpack会展开Ts里的每个类型
• 返回类型Tuple[Unpack[Tuple[Optional[T] for T in Ts]]]：遍历Ts里的每个类型T，用Optional[T]包装后，再通过Unpack组合成新的元组类型

示例验证 #

如果输入是tuple[dict[str, str], dict[str, str]]，类型检查器会正确推断返回类型为tuple[Optional[dict[str, str]], Optional[dict[str, str]]]，完全符合你的需求。

兼容旧版本Python #

如果使用Python 3.10及以下版本，需要先安装typing-extensions库，然后导入对应的类型：

from typing_extensions import TypeVarTuple, Unpack
from typing import Optional, Tuple

Ts = TypeVarTuple('Ts')

def f(tup: Tuple[Unpack[Ts]]) -> Tuple[Unpack[Tuple[Optional[T] for T in Ts]]]:
    return tuple(None if some_condition else el for el in tup)

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