numpy.ndarray类型提示/注释(PEP 484)?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (28)

是否有人实现了类型提示numpy.ndarray类?

现在,我在用typing.Any。

提问于
用户回答回答于

可以使用

In [586]: np.array({'test':1})   # a dictionary
Out[586]: array({'test': 1}, dtype=object)

In [587]: np.array(['one','two'])  # a list
Out[587]: 
array(['one', 'two'], 
      dtype='<U3')

In [589]: np.array({'one','two'})  # a set
Out[589]: array({'one', 'two'}, dtype=object)

然后:

def foo(x: np.ndarray) -> np.ndarray:

起作用了。当然,如果您的函数最终调用了numpy传递其参数的函数。asanyarray(与许多人一样),这样的注释将是不完整的,因为您的输入可能是list,或np.matrix

用户回答回答于

检查DataShape。它使用数据类型以及一些语法来确定输入和输出数组的大小。

扫码关注云+社区