Linux C调用Python 函数

1.Python脚本,名称为py_add.py

def add(a=1,b=1):
    print('Function of python called!')
    print('a = ',a)
    print('b = ',b)
    print('a + b = ',a+b)

2.C代码

#include <stdio.h>
#include <stdlib.h>
#include <Python.h>

int main(int argc,char **argv)
{
    //初始化,在使用python系统前,必须使用Py_Initialize对其进行初始化。它会载入Python的内建模块并添加系统路径到模块搜索路径中。
    Py_Initialize();
    //判断初始化是否成功
    if(!Py_IsInitialized())
        {
            printf("Python init failed!\n");
            return -1;
        }
    //PyRun_SimpleString 为宏,执行一段python代码
    //导入当前路径
    PyRun_SimpleString("import sys");
    PyRun_SimpleString("sys.path.append('./')");

    PyObject *pName = NULL;
    PyObject *pModule = NULL;
    PyObject *pDict = NULL;
    PyObject *pFunc = NULL;
    PyObject *pArgs = NULL;

    //加载名为py_add的python脚本
    pName = PyString_FromString("py_add");
    pModule = PyImport_Import(pName);
    if(!pModule)
        {
            printf("Load py_add.py failed!\n");
            getchar();
            return -1;
        }
    pDict = PyModule_GetDict(pModule);
    if(!pDict)
        {
            printf("Can't find dict in py_add!\n");
            return -1;
        }
    //找到函数名为add的函数    
    pFunc = PyDict_GetItemString(pDict,"add");
    if(!pFunc || !PyCallable_Check(pFunc))
        {
            printf("Can't find function!\n");
            getchar();
            return -1;
        }
    /*
    向Python传参数是以元组(tuple)的方式传过去的,
    因此我们实际上就是构造一个合适的Python元组就
    可以了,要用到PyTuple_New,Py_BuildValue,PyTuple_SetItem等几个函数
    */
    pArgs = PyTuple_New(2);

    //  PyObject* Py_BuildValue(char *format, ...) 
    //  把C++的变量转换成一个Python对象。当需要从 
    //  C++传递变量到Python时,就会使用这个函数。此函数 
    //  有点类似C的printf,但格式不同。常用的格式有 
    //  s 表示字符串, 
    //  i 表示整型变量, 如Py_BuildValue("ii",123,456)
    //  f 表示浮点数, 
    //  O 表示一个Python对象
    PyTuple_SetItem(pArgs,0,Py_BuildValue("i",123));
    PyTuple_SetItem(pArgs,1,Py_BuildValue("i",321));

    //调用python的add函数
    PyObject_CallObject(pFunc,pArgs);

    //清理python对象
    if(pName)
        {
            Py_DECREF(pName);
        }
    if(pArgs)
        {
            Py_DECREF(pArgs);
        }
    if(pModule)
        {
            Py_DECREF(pModule);
        }

    //关闭python调用
    Py_Finalize();

    return 0;

}

3.编译

gcc -I/usr/include/python2.7/  mian.c -o main -L/usr/lib/ -lpython2.7

备注:链接Python的库需要在最后,否则可能出现以下错误提示:
undefined reference to ‘Py_Initialize’

参考博客:https://blog.csdn.net/magictong/article/details/8947892

猜你喜欢

转载自blog.csdn.net/VioletHan7/article/details/82019804