python - 如何使用@cython装饰器声明np.ndarray类型

Question

我无法在任何地方找到专门解决此问题的任何文档:

以下代码:

cdef int foo( double data ):
    # ...
    return int( data )

可以写成:

@cython.returns(cython.int)
@cython.locals(data=cython.double)
def foo(data):
     return int(data)

但我无法找到等效的声明:

cdef foo(np.ndarray[double] data):

使用@cython.locals 会导致编译错误。

装饰器在 cython 中声明 numpy 数组的正确方法是什么？

Answer 1

这似乎没有在任何地方记录，但 memoryviews 被接受为一个字符串:

@cython.returns(cython.int)
@cython.locals(data="double[:]")
def foo(data):
     return int(data[0])

我想这通常可行，但是 numpy 数组的挑战是您必须cimport numpy，我认为这在纯 Python 代码中是不可能的。