pythonの型ライブラリを使用して複数の可能な型を指定する

Question

関数の型ヒントを指定したい場合は、をstrに、listをintにすることができます。 Pythonを使ってこれを行う方法はありますか？ような何か：

from typing import List 

def my_function(list_arg: List[str|int]): 

    ... 
    ... 

Answer 1

Unionは、このような状況のために正確に意図されます。それはあなたが指定したタイプのいずれかとすることができる新しいタイプを作成します。

from typing import Union, List 
def my_function(list_arg: Union[List[str], List[int]]) -> None: ... 

あなたは、多くの場合、それを使用する場合は、タイプエイリアスを作る：

MyList = Union[List[str], List[int]] 
def my_function(list_arg: MyList) -> None: ... 

私のように、一般的な機能を使用することをお勧めしませんあなたが何らかの理由でそれを必要としない限り、@ user6269244答えは示唆しています。 T = TypeVar('T', int, str)は単純なケースで動作しますが、より制限的です。

たとえば、あなたがこのようなものがあるとします。このコードでは

def apply_my_function(list_of_lists: List[MyList]) -> None: 
    # each item in list_of_lists is either a List[int] or List[str] 
    for arg in list_of_lists: 
    my_function(arg) 

を、あなたは何chioceを持っていない：あなたはUnionを必要とする（List[List[T]]あなたはint型のリストとリストの両方を保存させませんTはstrとintの間を同じ式で切り替えることができないため、strの値を変更する必要があります）。

しかし、あなたはmy_functionを呼び出すコードでUnionを使用するので、あなたがmy_function自身の署名でUnionを使用することもあります。

さらに、ジェネリックスは型の複雑さを増し、ある時点で型システムの制限に遭遇します。あなたは引数の型に依存するようにmy_functionの戻り値の型をしたいのであれば、それはたとえば、あなたが実際にそれらを必要とする場合のためにジェネリックを確保すると良いでしょう：

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

# this can't be achieved with Union: 
def my_function(list_arg: List[T]) -> T: ... 

Answer 2

from typing import TypeVar, List 

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

def my_function(list_arg: List[T]): 
...