Python类型提示：如何为可索引对象定义泛型类型？

首先说第一个问题：如果你需要提示一个支持整数索引提取元素的对象，Python标准库的typing.Sequence其实就是最适合的选择。它涵盖了所有常见的序列类型（比如list、tuple、str，甚至自定义的实现了序列协议的类），类型检查工具（比如mypy）能很好地识别它。用法示例：

from typing import Sequence

def get_first_element(data: Sequence[str]) -> str:
    return data[0]

不过如果你确实想自定义一个更精准的、只要求实现__getitem__（整数索引）的类型，那你遇到的错误是因为对Generic的用法理解有误——你直接把int和ReturnType作为Generic的参数，但Generic要求传入的必须是类型变量（TypeVar），不能是具体类型或者普通标识符。

正确的自定义泛型类型写法 #

先定义一个类型变量，再把它传给Generic：

from typing import TypeVar, Generic

# 定义一个类型变量，用来表示容器中元素的类型
T = TypeVar('T')

class IndexableContainer(Generic[T]):
    def __getitem__(self, key: int) -> T:
        ...  # 这里不需要实现具体逻辑，只是用于类型提示

这样就不会再抛出TypeError了。之后你就可以在函数中这样使用：

def pick_element(container: IndexableContainer[int], index: int) -> int:
    return container[index]

进阶：用抽象基类强化约束 #

如果你想让这个自定义类型更严谨，确保子类必须实现符合要求的__getitem__，可以结合abc.ABC和抽象方法：

from typing import TypeVar, Generic
from abc import ABC, abstractmethod

T = TypeVar('T')

class IndexableContainer(Generic[T], ABC):
    @abstractmethod
    def __getitem__(self, key: int) -> T:
        ...

这样任何继承IndexableContainer的类都必须实现接收int类型索引、返回T类型元素的__getitem__方法，类型检查工具也能更严格地校验。

最后补充一句：除非你有特殊的约束需求，否则直接用Sequence[T]会更省心，因为它已经覆盖了绝大多数你会遇到的可索引场景。

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