受约束的 TypeVar 和 Union 有什么区别？

Question

如果我想要一个可以是多种可能类型的类型，Unions 似乎是我的表达方式：

U = Union[int, str] 

U 可以是int 或str。

我注意到虽然TypeVars 允许可选的 var-arg 参数，但它们似乎也做同样的事情：

T = TypeVar("T", int, str)

T 和 U 似乎都只允许使用 str 和 int 类型。

这两种方式有什么区别，什么时候应该首选？

Answer 1

T 的类型必须在给定scope 内的多次使用中保持一致，而U 则不然。

使用Union 类型作为函数参数，参数和返回类型都可以不同：

U = Union[int, str]

def union_f(arg1: U, arg2: U) -> U:
    return arg1

x = union_f(1, "b")  # No error due to different types
x = union_f(1, 2)  # Also no error
x = union_f("a", 2)  # Also no error
x # And it can't tell in any of the cases if 'x' is an int or string

将其与参数类型必须匹配的TypeVar 进行比较：

T = TypeVar("T", int, str)

def typevar_f(arg1: T, arg2: T) -> T:
    return arg1

y = typevar_f(1, "b")  # "Expected type 'int' (matched generic type 'T'), got 'str' instead
y = typevar_f("a", 2)  # "Expected type 'str' (matched generic type 'T'), got 'int' instead

y = typevar_f("a", "b")  # No error
y  # It knows that 'y' is a string

y = typevar_f(1, 2)  # No error
y  # It knows that 'y' is an int

因此，如果允许多种类型，请使用TypeVar，但在单个范围内T 的不同用法必须相互匹配。如果允许多种类型，则使用Union，但在给定范围内U 的不同用法不需要​​相互匹配。