嵌套列表的 Python 类型注释

Question

我想注释我的返回类型，它恰好是一个包含整数列表的列表。这是注释:List[List[int]]好的？这是我的返回类型的确切示例:

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

Answer 1

是的，List[List[int]]是正确的类型。

作为旁注，当您不确定类型时，您可以定义该变量并使用 Mypy reveal_type方法让它猜测正确的类型。例如:

> cat foo.py
a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
reveal_type(a)

> mypy foo.py
1.py:2: note: Revealed type is 'builtins.list[builtins.list*[builtins.int]]'

这告诉你 a 的类型是 List[List[int]] .请注意 reveal_type不是一个有效的函数；它是 Mypy 中内置的一种特殊语法。如果您尝试运行 foo.py在 Python 中，它会抛出 NameError .

有关更多信息，请考虑阅读 Mypy docs .