如何注释 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 上作为候选版本可用
我可以在 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 上作为候选版本可用