NumPy ndarray dtype的类型提示？

Question

我希望一个函数在其dtype的同时包含NumPyndarray的类型提示。在

例如，有了列表，你可以做以下事情。。。在

def foo(bar: List[int]):
   ...

…为了给出类型提示，bar必须是由int组成的list

不幸的是，此语法对NumPyndarray抛出异常：

^{pr2}$

是否可以给dtype特定于np.ndarray的类型提示？在

Answer 1

您可以查看nptyping：

from nptyping import Array

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

或者您可以使用字符串作为类型提示：

^{pr2}$

Answer 2

据我所知，还不可能在函数签名的numpy数组类型提示中指定dtype。计划将来某个时候实施。有关当前开发状态的更多详细信息，请参见numpy GitHub issue #7370和numpy-stubs GitHub。在