这应该是一个微不足道的问题,但作为初学者并没有弄清楚。
我有以下使用 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...
提前致谢!