如何在Python中为自定义集合类型添加类型注解？

要让MyCollection[int]这种泛型形式被IntelliSense正确识别，你需要把自定义类定义为泛型类，同时实现Sequence抽象基类要求的必要方法，具体步骤如下：

1. 导入泛型相关工具 #

如果是Python 3.9及以上版本，可从collections.abc导入泛型工具；若要兼容更早版本，也可以从typing模块导入：

from collections.abc import Sequence, Generic
from typing import TypeVar

2. 定义类型变量 #

创建一个类型变量来代表集合中元素的类型，比如用T作为通用标识：

T = TypeVar('T')

3. 改造自定义集合类 #

让MyCollection同时继承Generic[T]和Sequence[T]，赋予类泛型能力；同时必须实现Sequence要求的抽象方法__getitem__和__len__（否则无法实例化）：

class MyCollection(Generic[T], Sequence[T]):
    def __init__(self, items: list[T] | None = None):
        self._items = items or []
    
    def __getitem__(self, index: int) -> T:
        return self._items[index]
    
    def __len__(self) -> int:
        return len(self._items)

4. 使用泛型类型注解 #

现在你可以正常使用MyCollection[int]这类泛型形式的类型注解，IntelliSense会正确识别：

obj: MyCollection[int] = MyCollection([1, 2, 3])
obj2: MyCollection[str] = MyCollection(["a", "b", "c"])

关键说明 #

• 继承Generic[T]是让类支持泛型参数的核心，它会告诉类型检查器（包括IntelliSense）这个类可以接受一个类型参数，用来指定集合内元素的类型。
• 实现__getitem__和__len__是因为Sequence是抽象基类，必须实现这两个方法才能创建类的实例。

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