python - Mypy 索引类型 "str"为 "Union[str, Dict[str, str]]"无效；预期类型 "Union[int, slice]"

Question

为什么我会收到错误消息？我已经正确添加了类型，对吗？

Invalid index type "str" for "Union[str, Dict[str, str]]"; expected type "Union[int, slice]"

代码

from typing import List, Dict, Union

d = {"1": 1, "2": 2}

listsOfDicts: List[Dict[str, Union[str, Dict[str, str]]]] = [
    {"a": "1", "b": {"c": "1"}},
    {"a": "2", "b": {"c": "2"}},
]

[d[i["b"]["c"]] for i in listsOfDicts]

Answer 1

Mypy 期望字典具有相同的类型。使用 Union 为子类型关系建模，但由于 Dict 类型是不变的，因此键值对必须完全匹配，如类型注释中定义的那样——这是 Union[str, Dict[str, str]] 类型，因此 Union 中的子类型不会匹配(str, Dict[str, str] 是有效类型)。

要为不同的键定义多种类型，您应该使用TypedDict。

用法如下所示:https://mypy.readthedocs.io/en/latest/more_types.html#typeddict .

from typing import List, Dict, Union, TypedDict

d = {"1": 1, "2": 2}

dictType = TypedDict('dictType', {'a': str, 'b': Dict[str, str]})

listsOfDicts: List[dictType] = [
    {"a": "1", "b": {"c": "1"}},
    {"a": "2", "b": {"c": "2"}},
]

[d[i["b"]["c"]] for i in listsOfDicts]