我正在将一个双数组从c函数传递给
python函数.我的代码是:
C代码:
double *compute(int size, const double a[])
{
double* array;
array = malloc(sizeof(double)*size);
for (int i=0; i
{
array[i] = 3*a[i];
}
//printf("Array in compute-function is: \n[");
//for(int i = 0; i < size; i++)
//printf("%f, ", array[i]);
//printf("]\n");
return array;
}
PYX代码:
cdef class ArrayWrapper:
cdef void* data_ptr
cdef int size
cdef set_data(self, int size, void* data_ptr):
""" Set the data of the array
This cannot be done in the constructor as it must recieve C-level
arguments.
Parameters:
-----------
size: int
Length of the array.
data_ptr: void*
Pointer to the data
"""
self.data_ptr = data_ptr
self.size = size
def __array__(self):
""" Here we use the __array__ method, that is called when numpy
tries to get an array from the object."""
cdef np.npy_intp shape[1]
shape[0] = self.size
# Create a 1D array, of length 'size'
ndarray = np.PyArray_SimpleNewFromData(1, shape,
np.NPY_INT, self.data_ptr)
return ndarray
def __dealloc__(self):
""" Frees the array. This is called by Python when all the
references to the object are gone. """
free(self.data_ptr)
def py_compute(int size, np.ndarray[np.double_t,ndim=1] a):
""" Python binding of the 'compute' function in 'GNLSE_RHS.c' that does
not copy the data allocated in C.
"""
cdef double *array
cdef np.ndarray ndarray
# Call the C function
array = compute(size, a.data)
array_wrapper = ArrayWrapper()
array_wrapper.set_data(size, array)
ndarray = np.array(array_wrapper, copy=False)
# Assign our object to the 'base' of the ndarray object
ndarray.base = array_wrapper
# Increment the reference count, as the above assignement was done in
# C, and Python does not know that there is this additional reference
Py_INCREF(array_wrapper)
return ndarray
Python代码:
for i in xrange(10):
x[i] = i;
a = cython_wrapper.py_compute(10, x)
print a
但我的结果是
[ 0 0 0 1074266112 0 1075314688 0 1075970048 0 1076363264]
而不是预期的
[ 0. 3. 6. 9. 12. 15. 18. 21. 24. 27.]
我的错误在哪里?我认为它与有问题的指针传输有关,但我不确定.