使用C++编写Python3扩展
时间:2022-07-28
本文章向大家介绍使用C++编写Python3扩展,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
Python的很多库中都包含了C/C++的代码,在安装这种库的时候,尝尝会遇到这样的报错:
error: unable to find vcvarsall.bat
通常的解决方案有两种:
- 安装visual studio或者mingw;
- 使用编译好的Python库文件,进行安装。
一般大家都会选择安装编译过的库文件,或者安装mingw,但是如果要在Windows下为自己的Python程序编写扩展,安装visual studio是最优的解决方案。
下面介绍使用C++编写Python扩展模块的常见写法。
传入有限个数据
先建立一个Exten.cpp文件,编写如下代码:
#include "Python.h"
// 函数主体
int add(int a,int b) {
return a + b;
}
// 包裹函数
static PyObject *Exten_add(PyObject *self,PyObject *args) {
int a,b;
// 获取数据,i代表int,ii代表两个int
// 如果没有获取到,则返回NULL
if (!PyArg_ParseTuple(args,"ii",&a,&b)) {
return NULL;
}
return (PyObject*)Py_BuildValue("i",add(a,b));
}
// 添加PyMethodDef ModuleMethods[]数组
static PyMethodDef ExtenMethods[] = {
// add:可用于Python调用的函数名,Exten_add:C++中对应的函数名
{"add",Exten_add,METH_VARARGS},
{NULL,NULL},
};
// 初始化函数
static struct PyModuleDef ExtenModule = {
PyModuleDef_HEAD_INIT,
"Exten",//模块名称
NULL,
-1,
ExtenMethods
};
void PyInit_Exten() {
PyModule_Create(&ExtenModule);
}
再在同文件夹下建立setup.py文件,编写如下代码:
from distutils.core import setup,Extension
MOD = 'Exten' #模块名
setup(name=MOD,ext_modules=[Extension(MOD,sources=['D:\Exten.cpp'])])
然后在CMD控制台输入:
python D:\setup.py build
python D:\setup.py install
此模块便已经安装成功了。可以直接在Python代码中调用:
>>> import Exten
>>> Exten.add(1,3)
4
传入可迭代对象
这种方式只能处理单个数据传入的任务,如果需要处理批量数组等批量数据,则需要涉及到Python对象到C++对象的转化。
以max函数为例,添加max函数之后,还需要为max编写包裹函数,整个Exten文件中代码如下:
#include "Python.h"
#include <vector>
#define INT_MIN -2147483648
using namespace std;
// 函数主体
int add(int a,int b) {
return a + b;
}
int max(vector<int> lst) {
int max_num = INT_MIN;
for (int i =0; i< lst.size(); i++) {
if (lst[i] > max_num) {
max_num = lst[i];
}
}
return max_num;
}
// 包裹函数
static PyObject *Exten_add(PyObject *self,PyObject *args) {
int a,b;
// 如果没有获取到数据
if (!PyArg_ParseTuple(args,"ii",&a,&b)) {
return NULL;
}
return (PyObject*)Py_BuildValue("i",add(a,b));
}
// 模块中每个可供Python调用的函数都需要一个对应的包裹函数
static PyObject *Exten_max(PyObject *self,PyObject *args) {
PyObject *obj;
vector<int> lst;
// O代表对象
if (!PyArg_ParseTuple(args,"O",&obj)) {
return NULL;
}
// 获取可迭代对象
PyObject *iter = PyObject_GetIter(obj);
if (!iter) {
PyErr_SetString(PyExc_TypeError,"The object is not iterable!");
return NULL;
}
while (true) {
// 逐个获取列表中的各个元素
PyObject *next = PyIter_Next(iter);
if (!next) {
break;
}
// 检查是否为long或者long的子类(包括int)
if (!PyLong_Check(next)) {
PyErr_SetString(PyExc_TypeError,"Int or Long list is expected!");
return NULL;
}
// 由Python的Long转化为C/C++的long
long num = PyLong_AsLong(next);
lst.push_back(num);
}
return (PyObject*)Py_BuildValue("i",max(lst));
}
// 添加PyMethodDef ModuleMethods[]数组
static PyMethodDef ExtenMethods[] = {
// add:可用于Python调用的函数名,Exten_add:C++中对应的函数名
{"add",Exten_add,METH_VARARGS},
{"max",Exten_max,METH_VARARGS},
{NULL,NULL},
};
// 初始化函数
static struct PyModuleDef ExtenModule = {
PyModuleDef_HEAD_INIT,
"Exten",//模块名称
NULL,
-1,
ExtenMethods
};
void PyInit_Exten() {
PyModule_Create(&ExtenModule);
}
采取同样的方式build + install之后,就可以使用了:
>>> import Exten
>>> Exten.max([1,2,3,4])
4
>>> Exten.max([1,2,3,4.1])
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: Int or Long list is expected!
参考:https://blog.csdn.net/baidu_35085676/article/details/79518777
https://stackoverflow.com/questions/22458298/extending-python-with-c-pass-a-list-to-pyarg-parsetuple
https://docs.python.org/3/c-api/long.html
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- 论文研读-基于决策变量分析的大规模多目标进化算法
- 用python画 pareto front
- 一起来学演化计算-matlab基本数据结构struct
- 一起来学演化计算-matlab基本函数inf, isempty, round, floor, fix
- 一起来学演化计算-matlab基本函数randperm end数组索引
- 论文研读-基于决策变量聚类的大规模多目标优化进化算法
- 一起来学演化计算-matlab基本函数min
- 一起来学演化计算-matlab基本函数find
- 欧拉图和哈密顿图
- python 操作 txt 文件中数据教程[4]-python 去掉 txt 文件行尾换行
- java字符数组char[]和字符串String之间的转换
- python操作txt文件中数据教程[3]-python读取文件夹中所有txt文件并将数据转为csv文件
- python操作txt文件中数据教程[1]-使用python读写txt文件
- python循环删除列表元素常见错误与正确方法
- Python字符串,整型,浮点数相互转化