c++调用python返回字典

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

#include <iostream> 

#include <Python.h> 

using namespace std; 

void HelloWorld(); 

void Add(); 

void TestTransferDict(); 

void TestClass(); 

int main() 

{ 

    cout << "Starting Test..." << endl; 

    cout << "HelloWorld()-------------" << endl; 

    HelloWorld(); 

    cout << "Add()--------------------" << endl; 

    Add(); 

    cout << "TestDict-----------------" << endl; 

    TestTransferDict(); 

    cout << "TestClass----------------" << endl; 

    TestClass(); 

    system("pause"); 

    return 0; 

} 

//调用输出"Hello World"函数 

void HelloWorld() 

{ 

    Py_Initialize();              //使用python之前，要调用Py_Initialize();这个函数进行初始化 

    PyObject * pModule = NULL;    //声明变量 

    PyObject * pFunc = NULL;      //声明变量 

    pModule =PyImport_ImportModule("Test001");              //这里是要调用的Python文件名 

    pFunc= PyObject_GetAttrString(pModule, "HelloWorld");   //这里是要调用的函数名 

    PyEval_CallObject(pFunc, NULL);                         //调用函数,NULL表示参数为空 

    Py_Finalize();                //调用Py_Finalize,这个和Py_Initialize相对应的. 

} 

//调用Add函数,传两个int型参数 

void Add() 

{ 

    Py_Initialize(); 

    PyObject * pModule = NULL;     

    PyObject * pFunc = NULL;       

    pModule =PyImport_ImportModule("Test001");      //Test001:Python文件名 

    pFunc= PyObject_GetAttrString(pModule, "add");  //Add:Python文件中的函数名 

　　//创建参数: 

    PyObject *pArgs = PyTuple_New(2);               //函数调用的参数传递均是以元组的形式打包的,2表示参数个数 

    PyTuple_SetItem(pArgs, 0, Py_BuildValue("i", 5));//0---序号  i表示创建int型变量 

    PyTuple_SetItem(pArgs, 1, Py_BuildValue("i", 7));//1---序号 

　　//返回值 

    PyObject *pReturn = NULL; 

    pReturn = PyEval_CallObject(pFunc, pArgs);      //调用函数 

　　//将返回值转换为int类型 

    int result; 

    PyArg_Parse(pReturn, "i", &result);    //i表示转换成int型变量 

    cout << "5+7 = " << result << endl; 

    Py_Finalize();                 

} 

//参数传递的类型为字典 

void TestTransferDict() 

{ 

    Py_Initialize(); 

    PyObject * pModule = NULL;     

    PyObject * pFunc = NULL;       

    pModule =PyImport_ImportModule("Test001");      //Test001:Python文件名 

    pFunc= PyObject_GetAttrString(pModule, "TestDict"); //Add:Python文件中的函数名 

　　//创建参数: 

    PyObject *pArgs = PyTuple_New(1);  

    PyObject *pDict = PyDict_New();   //创建字典类型变量 

    PyDict_SetItemString(pDict, "Name", Py_BuildValue("s", "WangYao")); //往字典类型变量中填充数据 

    PyDict_SetItemString(pDict, "Age", Py_BuildValue("i", 25));         //往字典类型变量中填充数据 

    PyTuple_SetItem(pArgs, 0, pDict);//0---序号  将字典类型变量添加到参数元组中 

　　//返回值 

    PyObject *pReturn = NULL; 

    pReturn = PyEval_CallObject(pFunc, pArgs);      //调用函数 

　　//处理返回值: 

    int size = PyDict_Size(pReturn); 

    cout << "返回字典的大小为: " << size << endl; 

    PyObject *pNewAge = PyDict_GetItemString(pReturn, "Age"); 

    int newAge; 

    PyArg_Parse(pNewAge, "i", &newAge); 

    cout << "True Age: " << newAge << endl; 

    Py_Finalize();                 

} 

//测试类 

void TestClass() 

{ 

    Py_Initialize(); 

    PyObject * pModule = NULL;     

    PyObject * pFunc = NULL;       

    pModule =PyImport_ImportModule("Test001");      //Test001:Python文件名 

    pFunc= PyObject_GetAttrString(pModule, "TestDict"); //Add:Python文件中的函数名 

　　//获取Person类 

    PyObject *pClassPerson = PyObject_GetAttrString(pModule, "Person"); 

    //创建Person类的实例 

    PyObject *pInstancePerson = PyInstance_New(pClassPerson, NULL, NULL); 

    //调用方法 

    PyObject_CallMethod(pInstancePerson, "greet", "s", "Hello Kitty");   //s表示传递的是字符串,值为"Hello Kitty" 

    Py_Finalize();           

}

1

2

3

4

5

6

7

8

9

10

11

12

13

14

#test.py 

def HelloWorld(): 

    print "Hello World" 

def add(a, b): 

    return a+b 

def TestDict(dict): 

    print dict 

    dict["Age"] = 17 

    return dict 

class Person: 

    def greet(self, greetStr): 

        print greetStr 

#print add(5,7) 

#a = raw_input("Enter To Continue...")