Python中使用ctypes调用C++的方法

int add(int a, int b) {
    return a+b;
}

extern "C"
int add(int a, int b);

$ g++ -fPIC -shared -o add.so add.cpp

from ctypes import *
lib = cdll.LoadLibrary('./add.so')
a = 5
b = 4
c = lib.add(a,b)
print("a + b = ", c)

void add_arr(double a[10][10], double b[10][10], double c[10][10]) {
    for (int i=0; i<10; i++) {
        for (int j=0; j<10; j++) {
            c[i][j] = a[i][j] + b[i][j];
        }
    }
}

extern "C"
void add_arr(double a[10][10], double b[10][10], double c[10][10]);

from ctypes import *
def double_ptr(arr):
    ptr = arr.ctypes.data_as(POINTER(c_double))
    return ptr
lib = cdll.LoadLibrary('./add_arr.so')
lib_add_arr = lib.add_arr
lib_add_arr.argtypes = [     #指定函数调用的参数形式
    POINTER(c_double),         #ctypes定义下的指向64位浮点数的指针
    POINTER(c_double), 
    POINTER(c_double) ]
a = np.random.rand(10,10)
b = np.random.rand(10,10)
c = np.zeros([10,10])
#调用函数，执行数据的相加
lib_add_arr(double_ptr(a), double_ptr(b), double_ptr(c))
print("a = \n", a)
print("b = \n", b)
print("c = \n", c)