Pybind11实现Python调用C++分三步:写C++函数、用PYBIND11_MODULE封装、CMake编译为.so/.pyd后import;关键在安全封装,如自动类型转换、避免裸指针、支持NumPy零拷贝。

用 pybind11 让 Python 调用 C++ 代码,核心就三步:写好 C++ 函数、用 pybind11 封装成 Python 模块、编译后在 Python 中 import 使用。关键不在“能不能”,而在“怎么封装得干净、安全、易维护”。
确保已安装 CMake(≥3.14)和 Python 的 pybind11 包:
pip install pybind11
项目结构建议简洁清晰:
不要直接暴露裸指针或全局状态。用 pybind11 的 py::class_、py::vector 等自动处理类型转换。例如实现一个支持 int/float 的加法:
立即学习“Python免费学习笔记(深入)”;
#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
<p>double add(double a, double b) {
return a + b;
}</p><p>// 绑定函数,名字在 Python 里叫 'add_py'
PYBIND11_MODULE(example, m) {
m.doc() = "pybind11 example plugin";
m.def("add_py", &add, "A function that adds two numbers");
}</p>注意:PYBIND11_MODULE 宏名(这里是 example)就是你将来 import 的模块名;函数名 add_py 是 Python 侧看到的名字。
在 CMakeLists.txt 中启用 pybind11 支持:
cmake_minimum_required(VERSION 3.14) project(example) <p>find_package(pybind11 REQUIRED) pybind11_add_module(example add.cpp)</p>
然后终端执行:
mkdir build && cd build cmake .. && make -j
成功后会在 build/ 下生成 example.cpython-*.so(Linux/macOS)或 example.pyd(Windows),重命名为 example.so 或直接用原名即可。
在 Python 文件中直接 import 并测试:
import example print(example.add_py(2.5, 3.7)) # 输出 6.2
想验证是否真提速?对比纯 Python 循环和 C++ 实现的向量求和:
std::vector + for loop,绑定为 sum_vector
sum([x for x in ...]) 和 example.sum_vector(...) 对比耗时记得传入 NumPy 数组时,在 C++ 端用 pybind11::array_t<double></double> 接收,它能零拷贝访问内存,避免复制开销。
以上就是如何使用pybind11让Python调用c++++代码 实现高性能混合编程【教程】的详细内容,更多请关注php中文网其它相关文章!
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号