関数に、NumPy ndarray
の型ヒントとそのdtype
の型ヒントを含めたいです。
たとえば、リストを使用すると、次のことができます...
def foo(bar: List[int]):
...
...bar
がlist
で構成されるint
でなければならないという型のヒントを与えるため。
残念ながら、この構文はNumPy ndarray
に対して例外をスローします:
def foo(bar: np.ndarray[np.bool]):
...
> np.ndarray[np.bool]) (...) TypeError: 'type' object is not subscriptable
dtype
固有の型ヒントをnp.ndarray
に与えることは可能ですか?
私の知る限りでは、関数のシグネチャでnumpy配列型のヒントにdtype
を指定することはまだ不可能です。将来的には実装される予定です。現在の開発ステータスの詳細については、 numpy GitHub issue#737 および numpy-stubs GitHub を参照してください。
あなたはチェックアウトすることができます nptyping :
from nptyping import Array
def foo(bar: Array[np.bool]):
...
または、型ヒントに文字列を使用することもできます。
def foo(bar: 'np.ndarray[np.bool]'):
...