Python中如何为SortedDict定义正确的类型注解避免下标报错

错误原因 #

你遇到的TypeError: 'type' object is not subscriptable报错，核心是Python 3.9之前的版本中，sortedcontainers库的SortedDict类没有实现泛型下标所需的__class_getitem__方法，运行时直接写SortedDict[int, str]会被当做对类对象执行下标取值操作，因此触发报错。
另外你示例代码中还有一个隐性问题：不能直接将普通dict赋值给SortedDict类型的变量，必须显式调用SortedDict构造方法传入字典数据，否则就算类型注解不报错，实际运行时变量类型也不匹配。

可行解决方案 #

• 方案1：使用__future__注解延迟求值（Python 3.7+适用）
  在代码第一行导入annotations，会将所有类型注解转为字符串存储，运行时不会执行下标操作，同时静态类型检查工具（mypy、pyright等）可以正常识别泛型类型：

from __future__ import annotations

from typing import Dict
from sortedcontainers import SortedDict

x: Dict[int, str] = {2: "two"}
y: SortedDict[int, str] = SortedDict({3: "three"})

• 方案2：使用注释式类型注解（全版本适用）
  如果需要兼容Python 3.7之前的版本，可以把类型注解写在变量赋值的注释中，运行时完全不会处理注解内容：

from typing import Dict
from sortedcontainers import SortedDict

x: Dict[int, str] = {2: "two"}
y = SortedDict({3: "three"})  # type: SortedDict[int, str]

• 方案3：升级到Python 3.9+并更新依赖
  Python 3.9开始支持标准库集合类型的泛型下标，最新版的sortedcontainers也已经适配了该特性，升级后可以直接使用泛型下标写法，只需要修正赋值逻辑即可：

from sortedcontainers import SortedDict

y: SortedDict[int, str] = SortedDict({3: "three"})

• 额外注意：静态类型检查适配
  如果使用mypy、pyright等工具做类型校验，需要安装对应的类型存根包保证工具能正确识别SortedDict的泛型定义：
  pip install types-sortedcontainers

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