Python 类型提示列表与可能的值

Question

我有一个 function：

def foo(a, b):
  return [a, b]

我想为返回值添加类型提示，例如，您可以看到我的 function 可以返回[srt, int]或[str, str]或[int, int]等。

我试过了：

def foo(a, b) -> List[str, int]:
  return [a, b]

但它不起作用。

如何在我的 function 可以返回的列表中指定可能的值？

Answer 1

您可以使用Union 。 具体来说， Union[X, Y]表示 X 或 Y。

from typing import List, Union

def foo(a, b) -> List[Union[str, int]]:
    return [a, b]

如果您的 function 可以返回包含任何元素的列表，则使用Any （表示无约束类型的特殊类型）：

from typing import Any, List

def foo(a, b) -> List[Any]:
    return [a, b]