递归类型别名在赋值语句中触发mypy错误的原因排查

错误原因 #

1. 类型窄化失效：mypy无法通过isinstance(x[0], float)自动推断出x的类型是list[float]。你的MatList是递归联合类型，mypy对这种嵌套联合类型的反向类型推导支持有限——它仅能识别x[0]是float，但没法确定整个列表的元素都是float，因此无法完成类型窄化。

2. 变量赋值不兼容：虽然list[list[float]]逻辑上属于MatList（因为list[float]是MatList，所以list[MatList]包含list[list[float]]），但mypy处理递归联合类型时，无法自动完成这种嵌套类型的兼容推导，因此触发赋值错误。

修复方案 #

方案1：使用自定义类型守卫函数 #

通过类型守卫明确告知mypy，当x[0]是float时，x属于list[float]：

from typing import TypeAlias, TypeGuard, cast

MatList: TypeAlias = list["MatList"] | list[float]

def is_float_list(x: MatList) -> TypeGuard[list[float]]:
    return isinstance(x[0], float)

def convert(x: MatList) -> MatList:
    if is_float_list(x):
        reveal_type(x)  # 现在会显示 "builtins.list[builtins.float]"
        x[0] = 1.2
    return x

# 用cast让mypy明确识别赋值类型
a: MatList = cast(MatList, [[1.0, 2.0], [3.0, 4.0]])
print(convert(a))

方案2：改用非递归的层级类型定义 #

如果你的场景是固定深度的嵌套（比如二维列表），可以避免递归联合类型，改用明确的层级定义：

from typing import TypeAlias, List

FloatList: TypeAlias = List[float]
MatList: TypeAlias = List[FloatList] | FloatList

def convert(x: MatList) -> MatList:
    if isinstance(x[0], float):
        reveal_type(x)  # 显示 "builtins.list[builtins.float]"
        x[0] = 1.2
    return x

a: MatList = [[1.0, 2.0], [3.0, 4.0]]  # 现在不会触发赋值错误
print(convert(a))

关键说明 #

• 递归联合类型虽语法合法，但mypy对其类型推导和窄化的支持不如非递归类型完善，能明确层级的场景下建议优先使用非递归定义。
• 类型守卫是解决复杂类型窄化问题的通用手段，可直接向类型检查器声明变量的具体类型。

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