主页 / coding / 问题 / 78800160
Accepted
bzm3r
Asked: 2024-07-27 07:31:04 +0800 CST

如何注释用文字初始化的“OrderedDict”的类型？

  • 772

假设我有以下内容：

from collections import OrderedDict
from dataclasses import dataclass


@dataclass
class HelloWorld:
    x: OrderedDict[str, int]


a = OrderedDict([("a", 0), ("c", 2), ("b", 1)])
HelloWorld(a) <--- # type error here

产生的类型错误是：

Argument of type "OrderedDict[Literal['a', 'c', 'b'], Literal[0, 2, 1]]" cannot be assigned to parameter "x" of type "OrderedDict[str, int]" in function "__init__"
  "OrderedDict[Literal['a', 'c', 'b'], Literal[0, 2, 1]]" is incompatible with "OrderedDict[str, int]"
    Type parameter "_KT@OrderedDict" is invariant, but "Literal['a', 'c', 'b']" is not the same as "str"
    Type parameter "_VT@OrderedDict" is invariant, but "Literal[0, 2, 1]" is not the same as "int

奇怪的是，这个非常相似的代码片段不会产生错误：

from collections import OrderedDict
from dataclasses import dataclass


@dataclass
class HelloWorld:
    x: OrderedDict[str, int]


HelloWorld(OrderedDict([("a", 0), ("c", 2), ("b", 1)])) # <--- no error
python

2 个回答

  1. Best Answer

    cast通常是一个可行的解决方案，但在这里您只需注释初始变量即可。这将保留代码的类型安全性（与cast可以同样好地强制转换str为不同dict[int, int]）。Pyright 不应该抱怨以下内容：

    from collections import OrderedDict
from dataclasses import dataclass

@dataclass
class HelloWorld:
    x: OrderedDict[str, int]

a: OrderedDict[str, int] = OrderedDict([("a", 0), ("c", 2), ("b", 1)])
HelloWorld(x=a)

    这里有一个游乐场。

    这个错误是由于对 和不变性 (如同)pyright一起的过于乐观的推断引起的。OrderedDictMutableMapping

    正如 @InSync 在评论中注意到的那样，在 Python 3.7 及更高版本中，您可以从 dict 文字构建 OrderedDict 并仍然保留顺序，并且在这种情况下推理不太具体：

    a = OrderedDict({"a": 0, "c": 2, "b": 1})
HelloWorld(x=a)
    • 3

  2. 使用typing.cast：

    from collections import OrderedDict
from dataclasses import dataclass
from typing import cast


@dataclass
class HelloWorld:
    x: OrderedDict[str, int]


a = OrderedDict([("a", 0), ("c", 2), ("b", 1)])
HelloWorld(cast(OrderedDict[str, int], a))
    • 0

相关问题