Python类型标注中集合的协变与不变性含义及问题解决方案咨询

问题场景 #

当在Python中使用泛型类型注解时，list的不变特性会触发类型检查报错。示例代码如下：

class A:
    pass
   
class B(A):
    pass

def print_list_a(my_list: list[A]) -> None:
    print(my_list)

l1 = [A()]
l2 = [B()]

print_list_a(l1)
print_list_a(l2) 

在严格模式下运行mypy时会得到错误提示：

main.py:41: error: Argument 1 to "print_list_a" has incompatible type "list[B]"; expected "list[A]"  [arg-type]
main.py:41: note: "List" is invariant
main.py:41: note: Consider using "Sequence" instead, which is covariant

核心概念解释 #

不变类型（Invariant） #

list[T]属于不变类型：即便B是A的子类，list[B]也不能被当作list[A]使用，反之亦然。

这种设计源于类型安全考量：列表是可变容器，如果允许list[B]隐式转换为list[A]，函数内部可能会向列表中添加A的其他子类甚至A本身的实例，导致原本仅包含B的列表混入不符合类型的元素，破坏类型一致性。

协变类型（Covariant） #

Sequence[T]属于协变类型：如果B是A的子类，那么Sequence[B]自动成为Sequence[A]的子类型，可以直接替代使用。

协变的安全前提是容器仅提供读取操作：当你只从容器中读取元素时，所有B的实例都可以被当作A的实例处理，不会出现类型安全问题。

最佳解决方案 #

方案1：改用Sequence[A]作为参数类型 #

如果函数仅对集合执行读取操作（如示例中的打印），将参数类型从list[A]改为Sequence[A]是最优解。Sequence是Python标准库定义的只读抽象接口，list、tuple等容器都实现了该接口，同时支持协变特性。

修改后的代码：

from typing import Sequence

class A:
    pass
   
class B(A):
    pass

def print_list_a(my_list: Sequence[A]) -> None:
    print(my_list)

l1 = [A()]
l2 = [B()]

print_list_a(l1)
print_list_a(l2)  # mypy检查通过

方案2：显式类型转换（仅适用于需要修改列表的场景） #

如果函数需要对列表执行修改操作（如添加元素），则必须保持list[A]的类型注解。此时可以通过typing.cast显式转换类型，但这种方式需谨慎使用——若函数内部真的添加了非B类型的元素，会导致运行时类型不一致问题。

示例代码：

from typing import cast

# ... 类定义和函数定义保持不变 ...

print_list_a(cast(list[A], l2))  # 显式转换后mypy检查通过

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