这些是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.floatpython内置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/

10-11 23:10
查看更多