如何注释 python 方法参数以指示正在传入 numpy 数组?

How to annotate a python method parameter to indicate a numpy array is being passed in?

我可以在 Python 3:

def foo(bar: int):
    pass

所以 bar 保证是 int

假设我有一个像这样的 np 数组:

bar = np.zeros(5, dtype=np.float32)

如何注释 foo 方法来强制执行此类型?

您可以使用 np.ndarray 指定您想要的数组。我认为 Numpy 目前不允许您使用泛型指定数据类型。请注意,这仅适用于 Numpy 1.20。

Numpy 1.20 在 PyPI 上作为候选版本可用