6

当我尝试运行下面的 cython 代码来生成一个空数组时，它会出现段错误。

有什么方法可以在 python 中生成空的 numpy 数组而不调用np.empty()？

cdef np.npy_intp *dims = [3]
cdef np.ndarray[np.int_t, ndim=1] result = np.PyArray_EMPTY(1, dims, 
                                                            np.NPY_INTP, 0)
4

1 回答 1

3

您可能很久以前就解决了这个问题，但是为了任何在试图弄清楚为什么他们的 cython 代码段错误时偶然发现这个问题的人的利益，这里有一个可能的答案。

当您在使用 numpy C API 时遇到段错误时，首先要检查的是您是否调用了该函数import_array()。这可能是这里的问题。

例如，这里是foo.pyx：

cimport numpy as cnp


cnp.import_array()  # This must be called before using the numpy C API.

def bar():
    cdef cnp.npy_intp *dims = [3]
    cdef cnp.ndarray[cnp.int_t, ndim=1] result = \
        cnp.PyArray_EMPTY(1, dims, cnp.NPY_INTP, 0)
    return result

这是setup.py构建扩展模块的简单方法：

from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
import numpy as np


setup(cmdclass={'build_ext': build_ext},
      ext_modules=[Extension('foo', ['foo.pyx'])],
      include_dirs=[np.get_include()])

这是正在运行的模块：

In [1]: import foo

In [2]: foo.bar()
Out[2]: array([4314271744, 4314271744, 4353385752])

In [3]: foo.bar()
Out[3]: array([0, 0, 0])
于 2014-12-14T01:01:32.780 回答