Python对象列表传入C++基类std::vector失败问题及解决方案咨询

问题核心 #

PyBind11对单个派生类对象到基类指针的转换做了原生支持，但对于std::vector<Base*>这类容器，默认不会自动将Python列表中的派生类实例隐式转换为基类指针容器，导致类型不匹配错误。

可行解决方案 #

方案1：绑定层封装适配函数（无需修改C++业务代码） #

在PyBind11绑定processAll时，直接接收Python列表，手动将每个元素转换为Base*后再调用C++方法：

// 修改PyProcessor的processAll绑定逻辑
.def("processAll", [](Processor& p, py::list py_list) {
    std::vector<Base*> cpp_vec;
    cpp_vec.reserve(py_list.size());
    // 遍历Python列表，逐个转换为Base*
    for (auto item : py_list) {
        cpp_vec.push_back(item.cast<Base*>());
    }
    p.process(cpp_vec);
}, PY_RVP_REFERENCE, PY_HELP("Process a list of Base-derived objects"));

Python端无需额外处理，直接传入包含派生类对象的列表即可：

l = [p1, p2]
processor.processAll(l)

方案2：改用智能指针容器（推荐，符合现代C++规范） #

如果可以修改C++代码，将容器类型改为std::vector<std::unique_ptr<Base>>，PyBind11对智能指针的多态支持更完善：

1. 更新C++的Processor类：

class Processor {
public:
    void process(std::vector<std::unique_ptr<Base>> v) {};
    void process(std::unique_ptr<Base> p) {}
};

2. 绑定类时指定智能指针作为持有者：

#include <pybind11/stl.h>

// 绑定Base类，指定std::unique_ptr作为持有者
py::class_<Base, std::unique_ptr<Base>>(m, "Base")
    .def(py::init<>());

// 绑定派生类时，指定继承关系和自身的智能指针类型
py::class_<Poly1, Base, std::unique_ptr<Poly1>>(m, "Poly1")
    .def(py::init<>())
    .def("polymethod", &Poly1::polymethod);

py::class_<Poly2, Base, std::unique_ptr<Poly2>>(m, "Poly2")
    .def(py::init<>())
    .def("polymethod", &Poly2::polymethod);

// 直接绑定processAll方法，无需额外转换
.def("processAll", &Processor::process, PY_RVP_REFERENCE, PY_HELP("Do something"));

Python端使用方式不变，列表中的派生类对象会自动被转换为std::unique_ptr<Base>存入容器。

方案3：自定义PyBind11类型转换器 #

通过注册自定义类型转换器，让PyBind11自动识别并转换Python列表到std::vector<Base*>：

#include <pybind11/stl.h>
namespace py = pybind11;

// 为std::vector<Base*>注册类型转换器
namespace pybind11::detail {
    template <> struct type_caster<std::vector<Base*>> {
    public:
        PYBIND11_TYPE_CASTER(std::vector<Base*>, _("List[Base]"));

        // 从Python对象加载到C++类型
        bool load(py::handle src, bool convert) {
            if (!py::isinstance<py::list>(src)) return false;
            py::list lst = src.cast<py::list>();
            value.reserve(lst.size());
            for (auto item : lst) {
                Base* ptr = item.cast<Base*>();
                if (!ptr) return false;
                value.push_back(ptr);
            }
            return true;
        }

        // 从C++类型转换到Python对象
        static py::handle cast(const std::vector<Base*>& src, py::return_value_policy policy, py::handle parent) {
            py::list lst;
            for (auto ptr : src) {
                lst.append(py::cast(ptr, policy));
            }
            return lst.release();
        }
    };
} // namespace pybind11::detail

// 正常绑定processAll方法即可
.def("processAll", &Processor::process, PY_RVP_REFERENCE, PY_HELP("Do something"));

该方案无需修改业务代码，通过扩展PyBind11的类型系统实现自动转换。

方案选择建议 #

• 优先选方案1：无需改动C++业务代码，实现成本最低。
• 推荐选方案2：符合现代C++内存管理规范，PyBind11支持更稳定。
• 追求通用型选方案3：适合需要频繁处理多态容器的场景。

内容的提问来源于stack exchange，提问作者Ben