小编Bod*_*rop的帖子

具有特定形状和数据类型的 Numpy 类型

目前,我正在尝试更多地使用 numpy 类型来使我的代码更清晰,但是我已经达到了目前无法覆盖的限制。

是否可以指定特定的形状以及相应的数据类型?例子:

Shape=(4,)
datatype= np.int32
Run Code Online (Sandbox Code Playgroud)

到目前为止,我的尝试如下所示(但都只是抛出错误):

第一次尝试:

import numpy as np

def foo(x: np.ndarray[(4,), np.dtype[np.int32]]):
...
result -> 'numpy._DTypeMeta' object is not subscriptable
Run Code Online (Sandbox Code Playgroud)

第二次尝试:

import numpy as np
import numpy.typing as npt

def foo(x: npt.NDArray[(4,), np.int32]):
...
result -> Too many arguments for numpy.ndarray[typing.Any, numpy.dtype[+ScalarType]]
Run Code Online (Sandbox Code Playgroud)

另外,不幸的是,我在文档中找不到有关它的任何信息,或者只有当我按照文档记录的方式实现它时才会出现错误。

numpy typing python-typing

29
推荐指数
1
解决办法
2万
查看次数

标签 统计

numpy ×1

python-typing ×1

typing ×1