这些是C文件:
添加一个.h
#ifndef __ADDONE
#define __ADDONE
void addone(float *in_data, int size);
#endif
添加一个.c
void addone(float *in_data, int size)
{
int i = 0;
for(int i = 0; i < size; i++)
{
in_data[i] = in_data[i] + 1;
}
}
我正试图将此函数与numpy中的ctypes一起使用:
import numpy as np
import numpy.ctypeslib as npct
from ctypes import c_int
array_1d_float = npct.ndpointer(dtype=np.float, ndim=1, flags="CONTIGUOUS")
libcd = npct.load_library("libaddone", ".")
libcd.addone.restype = None
libcd.addone.argtypes = [array_1d_float, c_int]
def addone(in_array):
return libcd.addone(in_array, len(in_array))
def main():
out = np.array([1,2,3], dtype=np.float)
print out
addone(out)
print out
if __name__ == "__main__":
main()
但是当我运行这个文件时,我得到了错误的结果:
python test.py
[1. 2. 3.]
[24.00000378 2.00000047 3. ]
怎么解决?
最佳答案
可以使用以下方法解决此问题:
void addone(double *in_data, int size)
而不是:
void addone(float *in_data, int size)
如https://stackoverflow.com/a/16964006中所述,
np.float
是python内置float
类型的别名,对应于c中的double
类型。在用
float
替换double
之前:$ python test.py
[ 1. 2. 3.]
[ 24.00000378 2.00000047 3. ]
将
float
替换为double
并重新编译库后:$ python test.py
[ 1. 2. 3.]
[ 2. 3. 4.]
或者,您可以保持c代码不变,并在python代码中使用
np.float32
而不是np.float
。关于python - 从numpy使用ctypeslib返回错误,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/37198473/