Numpy C ++:如何在没有段错误的情况下迭代PyArrayObject

时间:2013-09-13 15:18:30

标签: c++ python numpy python-c-api

对我来说,以下都会导致段错误:

my_array->descr->subarray->shape;
my_array->dimensions;
PyArray_SHAPE(my_array);
PyArray_DIMS(my_array);
PyArray_ITEMSIZE(my_array);
PyArray_NBYTES(my_array);

我的功能如下:

static PyObject* exterior(PyObject* self, PyArrayObject* old_simplices_array)
{//code here

我的cpp文件的其余部分如下所示:

#include "Python.h"
#include "numpy/arrayobject.h"

/* function */

static PyMethodDef compiled_methods[] = 
{
    {"_exterior",(PyCFunction)exterior , METH_VARARGS},
    {NULL,      NULL}        /* Sentinel */
};

PyMODINIT_FUNC init_alto(void)
{
    (void) Py_InitModule("_alto", compiled_methods);
    import_array();
}

将数组传递给“外部”的python代码只传递一个NxM uint数组。那部分有效。我可以访问数组的步幅和数据。我只是无法确定迭代的界限。如果这有任何不同,我在圣人内部工作。

如何在没有segfaulting的情况下迭代数组?如果答案很明显,请耐心等待答案。

为了更好地了解函数的外观,see here

1 个答案:

答案 0 :(得分:2)

过去我做了以下迭代PyArrayObject:

static PyObject *func1(PyObject *self, PyObject *args) {
    PyArrayObject *X;
    int ndX;
    npy_intp *shapeX;
    PyArray_Descr *dtype;
    NpyIter *iter;
    NpyIter_IterNextFunc *iternext;

    PyArg_ParseTuple(args, "O!", &PyArray_Type, &X);
    ndX = PyArray_NDIM(X);
    shapeX = PyArray_SHAPE(X);
    dtype = PyArray_DescrFromType(NPY_DOUBLE);

    iter = NpyIter_New(X, NPY_ITER_READONLY, NPY_KEEPORDER, NPY_NO_CASTING, dtype);
    if (iter==NULL) {
        return NULL;
    }
    iternext = NpyIter_GetIterNext(iter, NULL);
    dataptr = (double **) NpyIter_GetDataPtrArray(iter);

    do {
        cout << **dataptr << endl;
    } while (iternext(iter));

    NpyIter_Deallocate(iter);
    return Py_BuildValue(something);
}

要了解更多信息,请查看以下链接:http://docs.scipy.org/doc/numpy/reference/c-api.iterator.html