1

这应该是一个微不足道的问题,但作为初学者并没有弄清楚。

我有以下使用 numpy 数组的 Python C 扩展:

#include <cmath>
#include <Python.h>
#include <iostream>
#include "numpy/arrayobject.h"

using namespace std;
PyObject *func_calc(PyObject *self, PyObject *args)
{
    PyObject * PO_clmn;
    PyArrayObject * py_clmn;
    if (!PyArg_ParseTuple(args, "O", &PO_clmn))
        return NULL;
    py_clmn = (PyArrayObject*)PyArray_ContiguousFromObject(PO_clmn,PyArray_DOUBLE,1,1);
    double *clmn = (double*)(py_clmn->data);

    int i;
    int N = py_clmn->dimensions[0];
    int flag_threadholds[N_threadholds];
    for (i=0; i<N; i++)
    {
        clmn[i]=1;
    }
    return Py_None;
}

static PyMethodDef exampleMethods[] = 
{
    { "calc", func_calc, METH_VARARGS },
    { NULL, NULL }
} ;

PyMODINIT_FUNC initcalc()
{
    import_array();
    Py_InitModule("calc", exampleMethods);
}

将其编译为共享库后,发现以下调用未能将元素 clmn 数组修改为“1”:

import numpy
from calc import calc
clmn=numpy.zeros(10)
calc(clmn)
print clmn #[0,0...

提前致谢!

4

1 回答 1

1

根据您传入的数据,调用PyArray_ContiguousFromObject可能会返回原始对象,也可能会返回对象的副本。如果它返回一个副本,那么您的代码正在修改该副本,而不是原始对象。

于 2012-05-02T08:08:54.370 回答