Numpy，在 C++ 段错误中创建一维数组

Question

我正在尝试在 C++ 中为 numpy 创建一维数组：

PyObject* foo()
{
   npy_intp length[1];
   length[0] = 10;
   PyObject* my_array = PyArray_SimpleNew(1, length, NPY_DOUBLE);
   return boost::python::incref(my_array);
}

然后我做一个python模块：

BOOST_PYTHON_MODULE(pyMod)
{
    using namespace boost::python;

    import_array();

    def("foo", foo);
}

当我在 python 中运行它时，调用 foo()，它在执行对 PyArray_SimpleNew 的调用时会出现段错误，即使我已经调用了 import_array，这似乎是导致这种段错误的常见原因。如果我制作一个维度为 [10, 1] 的二维数组，则效果很好，即：

PyObject* foo()
{
   npy_intp length[2] = {10, 1};
   PyObject* my_array = PyArray_SimpleNew(2, length, NPY_DOUBLE);
   return boost::python::incref(my_array);
}

在python中是这样使用的

import numpy as np
import pyMod

if __name__ == "__main__":
    v = pyMod.foo()

对为什么会出现此段错误有任何想法吗？

Answer 1

我不确定您是如何将 PyObject* 暴露给 Python，但我认为正确的方法是通过 boost::python::object 暴露它。

另一点是您正在使用 PyObject 调用 boost::python::object。尽管有可能，但不建议这样做（参见 David Abrahams 撰写的 guidelines）。所以应该使用 boost::python::handle 来解决这种情况。

boost::python::object foo()
{
   npy_intp dims[1] = {10};
   PyObject* p_array = PyArray_SimpleNew(boost::size(dims), dims, NPY_FLOAT);
   boost::python::handle<> handle(boost::python::borrowed(p_array));
   return boost::python::object(handle);
}