Python递归类型及type(x)(...)的类型推断难题

问题重现 #

在尝试用递归类型标注嵌套数据结构时，会遇到MyPy类型推断的差异：

以下代码可正常通过MyPy检查：

IntType = int | list["IntType"] | tuple["IntType", ...]
StrType = str | list["StrType"] | tuple["StrType", ...]


def int2str(x: IntType) -> StrType:
    if isinstance(x, list):
        return list(int2str(v) for v in x)
    if isinstance(x, tuple):
        return tuple(int2str(v) for v in x)
    return str(x)

但合并list和tuple的类型判断后，看似等价的代码却报错：

IntType = int | list["IntType"] | tuple["IntType", ...]
StrType = str | list["StrType"] | tuple["StrType", ...]

def bad_int2str(x: IntType) -> StrType:
    if isinstance(x, (list, tuple)):
        return type(x)(bad_int2str(v) for v in x)  # 此处触发类型错误
    return str(x)

报错信息：

line 6: error: Incompatible return value type (
    got "list[int | list[IntType] | tuple[IntType, ...]] | tuple[int | list[IntType] | tuple[IntType, ...], ...]", 
    expected "str | list[StrType] | tuple[StrType, ...]"
)  [return-value]
line 6: error: Generator has incompatible item type 
    "str | list[StrType] | tuple[StrType, ...]"; 
    expected "int | list[IntType] | tuple[IntType, ...]"  [misc]

原因分析 #

这是MyPy的类型推断能力限制，而非代码逻辑错误，核心原因在于：

• 分开判断isinstance(x, list)和isinstance(x, tuple)时，MyPy能精确识别x的具体类型：比如判断为list时，它明确知道x是list[IntType]，因此能推断出list(int2str(v) for v in x)的结果是list[StrType]，完全匹配返回类型StrType。
• 但使用isinstance(x, (list, tuple))这种联合判断时，MyPy只能知道x属于list[IntType] | tuple[IntType, ...]，无法将type(x)的构造器与对应的目标递归类型建立关联。它会默认认为type(x)的构造器期望接收IntType类型的元素（原x的元素类型），但我们传入的生成器返回的是StrType，因此触发类型不匹配错误。

简言之，MyPy在联合类型场景下，无法跟踪type(x)对应的泛型参数转换关系，尤其是递归类型这种复杂场景，导致它没法推断出构造后的容器类型应该是list[StrType]或tuple[StrType, ...]。

解决方法 #

有两种可行的处理方式：

• 保持分开判断：像第一个函数那样，分别对list和tuple做类型检查，让MyPy能精确推断每个分支的返回类型，这是最清晰、无需额外类型断言的方案。
• 使用类型断言辅助：如果一定要合并判断，可以用cast明确告诉MyPy构造后的容器类型：

from typing import cast

IntType = int | list["IntType"] | tuple["IntType", ...]
StrType = str | list["StrType"] | tuple["StrType", ...]

def int2str(x: IntType) -> StrType:
    if isinstance(x, (list, tuple)):
        container = type(x)(int2str(v) for v in x)
        if isinstance(x, list):
            return cast(list[StrType], container)
        else:
            return cast(tuple[StrType, ...], container)
    return str(x)

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