Python 类型提示联合

Question

我有一个带有两个参数的函数：

def same_type_params(param1: Union[str, int], param2: Union[str, int]):
    pass

如何限制 和 的类型param1相等param2？即两者str或两者int

Answer 1

使用类型变量：

from typing import TypeVar
T = TypeVar('T', str, int)
def same_type_params(param1: T, param2: T) -> None:
    pass