numpy 如何将泛型类型添加到ndarray

0yycz8jy  于 2023-10-19  发布在  其他
关注(0)|答案(1)|浏览(92)

我有以下代码:

import numpy as np
from numpy import ndarray

def foo() -> ndarray:

    a = np.array([1.0, 2.2, 3.3333])
    print(a)
    print(type(a))

    return a

当我运行mypy时,我得到以下错误:

foo.py:5: error: Missing type parameters for generic type "ndarray"

我想在类型注解中指定以下信息:

  • 长度永远是3
  • 所有项目将浮动

这可能吗?

zzoitvuj

zzoitvuj1#

您需要以这种方式构建ndarray

a = np.ndarray(shape=3, dtype=float, buffer=np.array([1.0, 2.2, 3.3333]))

相关问题