需要修改代码C++BOOST_PYTHON_MODULE(Spline)
{
import_array();
boost::python::numeric::array::set_module_and_type("numpy", "ndarray");
class_("Spline", init<>())
.def("spline", &Spline::spline)
;
}
因此,在Python中,可以使用:from Spline.Spline import *
operation = Spline()
value, error_ = operation.spline(np.array(your_x_array), np.array(your_y_array), 0.01)
CPP:#define NUMBER_OF_SAMPLES 14
class Spline
{
public:
boost::python::list spline(numeric::array& x_val, numeric::array& y_val, double look_up_val);
};
然后进去boost::python::list Spline::spline(numeric::array& x_val, numeric::array& y_val, double p):PyArrayObject* x_pyArr = (PyArrayObject*)PyArray_FROM_O(x_val.ptr());
PyArrayObject* y_pyArr = (PyArrayObject*)PyArray_FROM_O(y_val.ptr());
int size = *(x_pyArr->dimensions), i , j;
double* data_x = (double*)x_pyArr->data;
double* data_y = (double*)y_pyArr->data;
double h[NUMBER_OF_SAMPLES], a, b, c, d, s[NUMBER_OF_SAMPLES] = { 0 }, F[NUMBER_OF_SAMPLES], f[NUMBER_OF_SAMPLES], x[NUMBER_OF_SAMPLES], m[NUMBER_OF_SAMPLES][NUMBER_OF_SAMPLES] = { 0 }, temp;
for (int i = 0; i < size; i++)
{
x[i] = *(data_x + i);
f[i] = *(data_y + i);
}
Spline::spline:boost::python::list return_val;
// ....
return_val.append(sum);
return_val.append(result);
return return_val;