我想将类型提示添加到将 numpy 数组作为输入并返回字符串的方法。这个 numpy 数组包含 float ,所以我尝试了:
import numpy as np
def foo(array: np.ndarray[np.float64]) -> str:
但由于 TypeError: 'type' object is not subscriptable
而无法工作。
我找到了 this但无法关注讨论!
最佳答案
查看 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.
关于python - Python 中的 Numpy 类型提示 (PEP 484),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52839427/