专栏首页IT派怎么给 Python 写 C 扩展?

怎么给 Python 写 C 扩展?

1. 环境准备

如果是Linux只需要安装Python3.x + Python-dev。

Windows下稍微复杂点,VS2017 + Python3.6.3

VS2017可用社区版,需要选择安装的环境如下:

2. Hello World !

2.1 C模块封装

以计算两个数相加为例,选择任意文件夹,新建如下C语言源码:

// 文件名 calc.c
#include 

int add(int x, int y){ // C 函数
 return x + y;
}

static PyObject *calc_add(PyObject *self, PyObject *args){

 int x, y;
 // Python传入参数
 // "ii" 表示传入参数为2个int型参数,将其解析到x, y变量中
 if(!PyArg_ParseTuple(args, "ii", &x, &y))
 return NULL;
 return PyLong_FromLong(add(x, y));
}

// 模块的方法列表
static PyMethodDef CalcMethods[] = {
 {"add", calc_add, METH_VARARGS, "函数描述"},
 {NULL, NULL, 0, NULL}
};

// 模块
static struct PyModuleDef calcmodule = {
 PyModuleDef_HEAD_INIT,
 "calc", // 模块名
    NULL, // 模块文档
 -1, /* size of per-interpreter state of the module,
                or -1 if the module keeps state in global variables. */
 CalcMethods
};

// 初始化
PyMODINIT_FUNC PyInit_calc(void)
{
 return PyModule_Create(&calcmodule);
}

其中,静态函数 calc_add 以python的C接口方式封装了add函数,命名方式 模块名_函数名

静态PyMethodDef列表 变量 CalcMethods 包含了该模块方法的描述

静态struct PyModuleDef结构体 变量 calcmodule 定义了模块的描述

PyInit_calc 函数初始化了模块,命名方式 PyInit_模块名

2.2 C源码编译

在VS2017中可以直接生成 .dll 文件,然后改名为 .pyd 就可在python程序中引入该模块了,但是,这不"清真",正确的姿势是写一个 setup.py然后通过python调cl.exe编译。

新建 setup.py文件,内容如下:

# setup.py

from distutils.core import setup, Extension

module1 = Extension('calc',
                    sources=['calc.c'])

setup(name='calc_model',
      version='1.0',
      description='Hello ?',
      ext_modules=[module1]
)

然后,从Windows的命令行(命令提示符)下进入到这个文件夹下,执行:

python setup.py build

即可完成编译,如果出现某 .bat文件未找到,说明你的VS没有安装相应的依赖(Linux下编译不成功原因可能是没有装python-dev),按文章开头给出的依赖库添加修改(此时不需要重新安装VS)。

编译结束后,在该文件夹下会出现 build 文件夹,进入该文件夹,出现如下两个文件夹:

进入 lib.xxx那个文件夹,里面有个 .pyd 结尾的文件(Linux下为 .so 结尾),这就是我们编译好的python模块了,如下:

当然,你也可以改名为 calc.pyd 比较好看,不过这不影响调用。

2.3 Python调用

这部分就简单了,进入含有编译好的 .pyd 文件夹,新建如下文件:

import calc

print(calc.add(12, 21))

这就是一个普通库,这样调用就OK了。

3. Python的参数传递以及C的返回值相关问题

这部分我直接甩出文件就行,编译及调用过程与上面一样。

C 文件

/**构建返回值
Py_BuildValue("")                        None
Py_BuildValue("i", 123)                  123
Py_BuildValue("iii", 123, 456, 789)      (123, 456, 789)
Py_BuildValue("s", "hello")              'hello'
Py_BuildValue("y", "hello")              b'hello'
Py_BuildValue("ss", "hello", "world")    ('hello', 'world')
Py_BuildValue("s#", "hello", 4)          'hell'
Py_BuildValue("y#", "hello", 4)          b'hell'
Py_BuildValue("()")                      ()
Py_BuildValue("(i)", 123)                (123,)
Py_BuildValue("(ii)", 123, 456)          (123, 456)
Py_BuildValue("(i,i)", 123, 456)         (123, 456)
Py_BuildValue("[i,i]", 123, 456)         [123, 456]
Py_BuildValue("{s:i,s:i}", "abc", 123, "def", 456)    {'abc': 123, 'def': 456}
Py_BuildValue("((ii)(ii)) (ii)", 1, 2, 3, 4, 5, 6)          (((1, 2), (3, 4)), (5, 6))
**/

#include


static PyObject *value_commonArgs(PyObject *self, PyObject *args){
 // 传入普通参数,例如: s = value.com(1, 2.3, "Hello C")
 int x;
 double y;
 char *z;
 if(!PyArg_ParseTuple(args, "ids", &x, &y, &z))
 return NULL;
    printf("The args is %d and %f and %s .n", x, y, z);
 // 返回(x, y, z)的元组
 return Py_BuildValue("(i,d,s)",x, y, z);
}



static PyObject *value_tupleTest(PyObject *self, PyObject *args){
 // t = value.tut((1, 3), "Tuple")
 int x, y;
 char *z;
 if(!PyArg_ParseTuple(args, "(ii)s", &x, &y, &z))
 return NULL;
    printf("The args is (%d, %d), %s .n", x, y, z);

 // return ([1, 2], "hello")
 return Py_BuildValue("[i,i]s", x, y, z);
}



static PyObject *value_some(PyObject *self, PyObject *args){
 /* 可选参数,可能是下面几种, "|" 代表后面的参数可选
        c = value.som(1)
        value.som(1, 3)
        value.som(1, 2, "hello")
    */
 int x = 0, y = 0;
 char *z = NULL;
 if(!PyArg_ParseTuple(args, "i|is", &x, &y, &z))
 return NULL;
    printf("x is: %dn", x);
    printf("y is: %dn", y);
 if(z != NULL)printf("z is: %sn", z);
 return 

本文分享自微信公众号 - IT派(transfer_3255716726)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2018-05-28

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 人脸融合?没有想象中难!

    颜如玉 —— python + opencv 人脸融合程序,可实现类似天天P图疯狂换脸、face++人脸融合效果

    IT派
  • 实战!视觉SLAM中的直接法在智能驾驶中的应用实践!

    来源:HorizonRobotics 本期由智能驾驶团队吴佳田、颜沁睿、杨德刚给大家分享他们在研发中对视觉SLAM直接法的应用实践,相关代码及说明文档已在Gi...

    IT派
  • 这里有10个优质Python开源项目,来帮你学好Python

    导读:在过去的一个月中,我们对近250个Python开源项目进行了排名,选出了前十。

    IT派
  • 自定义Android注解Part2:代码自动生成

    上一期我们已经把butterknife-annotations中的注解变量都已经定义好了,分别为BindView、OnClick与Keep。

    Rouse
  • JQ动态生成的元素,原事件绑定失效

    逸鹏
  • 分层风险平价:基于图论和机器学习的新资产配置方法(附代码)

    风险平价是构建多样化和均衡投资组合十分流行选择。众所周知,大多数资产类别的未来表现很难预测。 通过仅使用资产的风险特征和相关矩阵构建投资组合,风险平价方法克服了...

    量化投资与机器学习微信公众号
  • Java数据库编程--JDBC

    听着music睡
  • Metasploit框架MSFconsole命令详解

    MSFconsole核心命令教程 MSFconsole有许多不同的命令选项可供选择。以下是Metasploit命令的核心组合,并参考其格式。

    轩辕小子
  • 这是有赞的分层自动化测试实践

    先理一下自动化测试的概念,从广义上来说,一切通过工具(程序)的方式来代替或者辅助手工测试的行为都可以成为自动化。从狭义上来说,通过编写脚本的方式,模拟手工测试的...

    DevOps时代
  • Windows 10:系统美化篇

    Traffic Monitor 是一款主打「颜值」的网速监控小工具,具体效果如上图.

    乐心湖

扫码关注云+社区

领取腾讯云代金券