NumPy 类型提示某事物既是数组又是 float32？ [复制]

Posted 2023-03-22

【中文标题】NumPy 类型提示某事物既是数组又是 float32？ [复制]

【英文标题】：NumPy type hint that something is both an array and float32? [duplicate]

【发布时间】：2020-01-22 07:07:49

【问题描述】：

如何键入提示函数返回的值既是 NumPy 数组并保存 NumPy float32 数据？

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

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

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

我可以使用以下方法指定返回值的类型为 float32：

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

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

有没有办法在返回类型中指定两者？

【问题讨论】：

@Georgy，我同意它已经足够接近了。对于最初没有找到该问题，我深表歉意。

【参考方案1】：

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