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