NumPy 类型提示某些东西既是数组又是 float32？

Question

我如何键入提示 return 由函数编辑的值是 NumPy 数组并保存 NumPy float32 数据？

我可以指定 returned 值是一个数组，使用：

def func() -> np.ndarray:
    ...

但是，这并不强制知道它是一个 float32 数组。

我可以指定 returned 值是 float32 类型，使用：

def func() -> np.float32:
    ...

但是，这并不强制知道它是一个数组（与奇异标量值相反）。

有没有办法让我在 return 类型中指定两者？

Answer 1

目前这是不可能的，但这是一个受欢迎的功能请求；请参阅 this GitHub issue，其中包含有关进度和可能的解决方法的信息。

NumPy type hint that something is both an array and float32?