键入defaultdict时擦干

Question

考虑一下像这样的defaultdict：

my_dict = defaultdict(int)

太棒了。现在我想添加输入信息，以消除那些恼人的链接警告。类型已经部分确定为DefaultDict[Unknown, int]：缺少键的类型。

要键入它，我可以：

my_dict: DefaultDict[str, int] = defaultdict(int)

这重复了很多信息：DefaultDict和int类型，只是为了跨键获取信息。

有什么办法能使这件事更简洁吗？

Answer 1

您可以定义将值修正为int的泛型类型别名。

from typing import TypeVar

T = TypeVar('T')
IntMap = defaultdict[T, int]


my_dict: IntMap[str] = defaultdict(int)