Python中的Numpy类型提示（PEP 484）

Question

我想向采用numpy数组作为输入并返回字符串的方法中添加类型提示。这个numpy数组包含浮点数，所以我尝试了：

import numpy as np
def foo(array: np.ndarray[np.float64]) -> str:

但是由于TypeError: 'type' object is not subscriptable而无法使用。

我找到了this，但无法进行讨论！

我先谢谢你！

Answer 1

签出nptyping。它为numpy数组提供类型提示。

对于您而言，您最终会得到：

import numpy as np
from nptyping import Array

def foo(array: Array[np.float64]) -> str:
    ...

您也可以检查实例：

arr = np.array([[1.0, 2.0],
                [3.0, 4.0],
                [5.0, 6.0]])
isinstance(arr, Array[np.float64, 3, 2])  # True