我最近一直在尝试用C ++编写一个函数,它将双精度矢量转换为字符串矢量。我想从python解释器运行它,所以我使用Pybind11来连接C ++和Python。这是我到目前为止所做的,
#include <pybind11/pybind11.h>
#include <boost/lexical_cast.hpp>
#include <iostream>
#include <iomanip>
#include <algorithm>
#include <vector>
std::string castToString(double v) {
std::string str = boost::lexical_cast<std::string>(v);
return str;
}
std::vector<std::vector<std::string> > num2StringVector(std::vector<std::vector<double> >& inputVector) {
//using namespace boost::multiprecision;
std::vector<std::vector<std::string> > outputVector;
std::transform(inputVector.begin(), inputVector.end(), std::back_inserter(outputVector), [](const std::vector<double> &iv) {
std::vector<std::string> dv;
std::transform(iv.begin(), iv.end(), std::back_inserter(dv), &castToString);
return dv;
});
return outputVector;
}
namespace py = pybind11;
PYBIND11_PLUGIN(num2String) {
py::module m("num2String", "pybind11 example plugin");
m.def("num2StringVector", &num2StringVector, "this converts a vector of doubles to a vector of strings.");
m.def("castToString", &castToString, "This function casts a double to a string using Boost.");
return m.ptr();
}
现在,使用命令行中的follow命令编译成共享库:
c++ -O3 -shared -std=gnu++11 -I ../include `python-config --cflags --ldflags --libs` num2StringVectorPyBind.cpp -o num2String.so -fPIC -lquadmath
其中../include是pybind11包含的位置。编译完成后启动python并使用,
import num2String
values = [[10, 20], [30, 40]]
num2String.num2StringVector(values)
我得到以下错误,“不兼容的函数参数。支持以下参数类型”。然后它给了我一个可能的类型列表,这很奇怪,因为我只是想使用一个向量作为我的函数参数,根据pybind11的文档,这是一个支持的数据类型: http://pybind11.readthedocs.io/en/latest/basics.html#supported-data-types
是否我有一个矢量矢量(一个2d矢量)并且不受支持?
答案 0 :(得分:6)
您可以使用一行额外的代码来完成此工作:
#include <pybind11/stl.h>
支持开箱即用的基本数据类型(有些可能需要包含额外的扩展标题)。
在加入pybind11/stl.h
之前:
>>> import num2String
>>> num2String.num2StringVector([[1, 2], [3, 4, 5]])
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: Incompatible function arguments. The following argument types are supported:
1. (std::vector<std::vector<double, std::allocator<double> >, std::allocator<std::vector<double, std::allocator<double> > > >) -> std::vector<std::vector<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::allocator<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > > >, std::allocator<std::vector<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >, std::allocator<std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> > > > > >
包括pybind11/stl.h
并重新编译共享库:
>>> import num2String
>>> num2String.num2StringVector([[1, 2], [3, 4, 5]])
[['1', '2'], ['3', '4', '5']]