Как преобразовать ND-массив ND-массив в массив CFFI C ++ и обратно?

Я хочу передать массив NumPy в некоторый (один) код C ++ через CFFI. Предположим, я не могу (ни в каком смысле) изменить код C ++, интерфейс которого:

double CompactPD_LH(int Nbins, double * DataArray, void * ParamsArray ) {
...
}

Я передаю Nbins как целое число в Python, ParamsArray как структуру dict ->, но DataArray (shape = 3 x NBins, который должен быть заполнен из массива numpy, вызывает у меня головную боль. (Cast_matrix from Почему Cffi намного быстрее, чем NumPy? здесь не помогает 🙁

Вот одна попытка, которая не удалась:

from blah import ffi,lib
data=np.loadtxt(histof)
DataArray=cast_matrix(data,ffi) # see https://stackoverflow.com/questions/23056057/why-is-cffi-so-much-quicker-than-numpy/23058665#23058665
result=lib.CompactPD_LH(Nbins,DataArray,ParamsArray)

Для справки: cast_matrix был:

def cast_matrix(matrix, ffi):
ap = ffi.new("double* [%d]" % (matrix.shape[0]))
ptr = ffi.cast("double *", matrix.ctypes.data)
for i in range(matrix.shape[0]):
ap[i] = ptr + i*matrix.shape[1]
return ap

Также:

Как передать массив Numpy в функцию cffi и как вернуть его обратно?

https://gist.github.com/arjones6/5533938

1

Решение

Спасибо @morningsun!

dd=np.ascontiguousarray(data.T)
DataArray = ffi.cast("double *",dd.ctypes.data)
result=lib.CompactPD_LH(Nbins,DataArray,ParamsArray)

работает!

2

Другие решения

Других решений пока нет …

По вопросам рекламы [email protected]