我有一个C ++代码,它使用boost_python包执行python脚本。一切都很好,就像我从python中提取int,string或其他非数组变量一样。但是我必须提取一个numpy::ndarray
并将其转换为cpp vector
。我尝试如下:
main.cpp
#include <iostream>
#include <boost/python.hpp>
#include <boost/python/numpy.hpp>
using namespace boost::python;
int main()
double t_end=7
try
{
Py_Initialize();
object module = import("__main__");
object name_space = module.attr("__dict__");
exec_file("MyModule.py", name_space, name_space);
object MyFunc = name_space["MyFunc"];
object result = MyFunc(t_end);
auto result_array = extract<numpy::ndarray>(result);
const numpy::ndarray& ret = result_array();
int input_size = ret.shape(0);
double* input_ptr = reinterpret_cast<double*>(ret.get_data());
std::vector<double> v(input_size);
for (int i = 0; i < input_size; ++i)
v[i] = *(input_ptr + i);
}
catch (error_already_set)
{
PyErr_Print();
}
Py_Finalize();
例如py脚本:
MyModule.py
import numpy as np
def MyFunc(t_end):
result = np.array([2,3,1,t_end])
return results
但它以错误结束:
读取访问冲突BOOST_NUMPY_ARRAY_API为nullptr
我也试图像numpy::ndarray result_array = extract<numpy::ndarray>(result);
一样直接声明numpy :: ndarray但错误完全相同。我已经通过直接从python中打印来检查我的ndarray
是否为空,但事实并非如此。在python步骤似乎都是正确的。那么是什么导致违规以及如何解决?
由于您在没有首先初始化它的情况下使用numpy
模块,因此会发生该错误。
注意官方tutorial的开头:
初始化Python运行时和numpy模块。无法调用这些会导致分段错误:
namespace np = boost::python::numpy; int main(int argc, char **argv) { Py_Initialize(); np::initialize();
你的代码缺乏对np::initialize();
的调用。