使用c ++ boost :: python从python函数返回的列表中获取数据?

问题描述

我有一个返回浮点列表的python函数

# Script.py
def get_double_list():
    data_list = []
    for a in range(5):
        data_list.append(float(a/10));
    return data_list

我想从C ++代码调用它,并从列表中获取数据。 我编写了以下C ++代码,但提取双打不起作用。 我在做什么错了?

#define BOOST_AUTO_LINK_NOMANGLE
#include <boost/python.hpp>
#include <iostream>

namespace bp = boost::python;

int main(int argc,char** argv)
{

    Py_Initialize();
    boost::python::object main_module = boost::python::import("__main__");
    boost::python::dict main_namespace = boost::python::extract<boost::python::dict>(main_module.attr("__dict__"));

    exec_file("Script.py",main_namespace,main_namespace);

    boost::python::object global(main_module.attr("__dict__"));
    boost::python::object get_double_list = global["get_double_list"];

    if (!get_double_list.is_none())
    {
        auto py_ex_list = bp::extract<bp::list>(get_double_list());
        if (py_ex_list.check())
        {
            bp::list py_list = py_ex_list();
            Py_ssize_t list_len = bp::len(py_list);
            std::cout << "List length is: " << list_len << std::endl;
            for (Py_ssize_t i = 0; i < list_len; i++)
            {
                auto item = py_list[i];
                auto d_ex = bp::extract<double>(item);
                double d = d_ex();
                std::cout << "Double = " << d << std::endl;
            }
        }
    }
    return 0;
}

我知道它以某种方式失败了

auto d_ex = bp::extract<double>(item);

(已在MS Windows 10,Boost 1.73.0,Python 3.8.5上进行了测试)

解决方法

如果我将 data_list 更改为全局,则可以使用

data_list = []

def get_double_list():
    global data_list
    for a in range(5):
        data_list.append(float(a/10));
    return data_list