为NumPy的类型暗示ndarray D型？

Question

我想一个功能包括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型提示？

Answer 1

据我所知这是不可能的但在函数签名numpy的数组类型提示指定dtype。它计划在未来的某个时候实施。见numpy GitHub issue #7370和numpy-stubs GitHub对目前的发展状态的详细信息。

Answer 2

你可以检查出nptyping：

from nptyping import Array

def foo(bar: Array[np.bool]):
   ...

或者你可以只使用类型提示字符串：

def foo(bar: 'np.ndarray[np.bool]'):
   ...