用C编写Python扩展的一些代码模板

说明

用C语言编写.dll/.so为Python加速，有一些固定代码，2.x和3.x略有不同，记录如下。

2.x模板

#include "Python.h"

PyObject* get_fruit(PyObject *self, PyObject *args)
{
    return Py_None;
}

PyMethodDef module_methods[] = {
    { "get_fruit", get_fruit, METH_VARARGS, NULL },
    { NULL }
};

PyMODINIT_FUNC initmymod(void)
{
    Py_InitModule("mymod", module_methods);
}

3.x模板

#include "Python.h"

PyObject* get_fruit(PyObject *self, PyObject *args)
{
    return Py_None;
}

PyMethodDef module_methods[] = {
    { "get_fruit", get_fruit, METH_VARARGS, NULL },
    { NULL }
};
struct PyModuleDef cModPyDem ={ PyModuleDef_HEAD_INIT, "mymod", NULL, -1, module_methods };

PyMODINIT_FUNC PyInit_mymod(void)
{
    return PyModule_Create(&cModPyDem);
}

常用函数

• PyArg_ParseTuple Python创建get_fruit.args为一个tuple，所以先要用PyArg_ParseTuple解析

PyObject* arg1 = NULL;
if (PyArg_ParseTuple(args, "O", &arg1)==0){
        return Py_None;
}

• PyBytes_FromFormat

PyBytes_FromFormat("format");

• Py_BuildValue

Py_BuildValue("i", 123);

• PyErr_Format

PyErr_Format(PyExc_TypeError, "type error");

在C端使用numpy

1. 设置头文件路径：\Lib\site-packages\numpy\core\include
2. 包含头文件：#include "numpy/arrayobject.h"
3. 用PyArrayObject接收numpy.ndarray类型的数据

PyArrayObject常用成员变量说明

变量名	含义	类型
nd	数据是几维的，也就是dimensions数组的大小	int
dimensions	每个维度有几个元素	int64*
strides	每个维度内，访问下一个元素需要前移几个字节	int64*
data	原始数据的指针	char*

输出二维np.ndarray数据的代码段

get_data.py提供待处理数据

import numpy as np
def get_data():
    return np.array([ [10.0, 10.1, 10.2],[20.0, 20.1, 20.2],[30.0, 30.1, 30.2] ], dtype='float32')

C代码

inline float& dataof(PyArrayObject* obj, long long r, long long c)
{
    float* data = (float*)(obj->data);
    long long cols = obj->dimensions[0];
    return data[c*cols + r];
}

int main()
{
    Py_Initialize();

    PyObject *pModule = PyImport_ImportModule("get_data");
    PyObject *pDict = PyModule_GetDict(pModule);
    PyObject *pFunc = PyDict_GetItemString(pDict, "get_data");
    PyArrayObject* list = (PyArrayObject*)PyObject_CallObject(pFunc, nullptr);

    printf("nd=%d;dims=(%d,%d)\n", list->nd, int(list->dimensions[0]), int(list->dimensions[1]));
    for (int i = 0; i < list->dimensions[0]; i++)
    {
        for (int j = 0; j < list->dimensions[1]; j++)
        {
            printf("%f ", dataof(list, i, j));
        }
        printf("\n");
    }

    Py_Finalize();
    return 0;
}