Numpy C++: How to iterate over PyArrayObject without a segfault

Question

For me, the following all result in a segfault:

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

My function looks like this:

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

The rest of my cpp file looks like this:

#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();
}

The python code that passes the array to "exterior" just passes an NxM uint array. That part works. I can access the array's strides and data. I just cannot determine the bounds of iteration. I am working from within sage if that makes any difference.

How am I supposed to iterate over an array without segfaulting? If the answer is obvious, please idiotproof your answer.

For a better idea of what the function looks like, see here.

Joel Vroom · Accepted Answer · 2013-09-17 22:59:57Z

2

In the past I have done the following to iterate over a 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);
}

To find out more information check out this link: http://docs.scipy.org/doc/numpy/reference/c-api.iterator.html

edited Sep 17, 2013 at 22:59

answered Sep 13, 2013 at 16:06

Joel Vroom

1,6911 gold badge18 silver badges33 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Alex Eftimiades Over a year ago

I thought your example was working, but I was just getting a NULL iter. Are you sure you do not have to do anything special to the array that python passes to C?

Joel Vroom Over a year ago

I edited the above to show how X is parsed. This works for me. PyArg_ParseTuple(args, "O!", &PyArray_Type, &X);

Collectives™ on Stack Overflow

Numpy C++: How to iterate over PyArrayObject without a segfault

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related