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 数组提供类型提示。

在你的情况下，你最终会得到：

from nptyping import NDArray, Float64

def foo(array: NDArray[Float64]) -> str:
    ...

您也可以检查您的实例：

import numpy as np
from nptyping import NDArray, Float64

arr = np.array([[1.0, 2.0],
                [3.0, 4.0],
                [5.0, 6.0]])

isinstance(arr, NDArray[(3, 2), Float64])  # True.

# Or if you don't want to check the dimensions and their sizes:
isinstance(arr, NDArray[Float64])  # Also True.