使用 ctypes 将 C 数组从 C 函数返回到 Python

Question

我目前正试图通过编写一个 C 函数来完成一个非常大的数组的繁重工作，以减少 Python 程序的运行时间。 目前我只是在使用这个简单的功能。

int * addOne(int array[4])
{
    int i;
    for(i = 0; i < 5; i++)
    {
        array[i] = array[i] + 1;
    }
    return array;
}

我想让我的 Python 代码做的就是调用 C 函数，然后返回新数组。 这是我到目前为止所拥有的：

from ctypes import *
libCalc = CDLL("libcalci.so")
pyarr = [65, 66, 67, 68]
arr = (ctypes.c_int * len(pyarr))(*pyarr)
res = libCalc.addOne(arr)

如何从返回的指针创建 Python 列表？

Answer 1

您返回的指针实际上与您传递的指针相同。 即您实际上并不需要返回数组指针。

您将指向支持列表的内存区域的指针从 Python 移交给 C，然后 C 函数可以更改该内存。 您可以返回一个整数状态代码来标记是否一切都按预期进行，而不是返回指针。

int addOne(int array[4])
{
    int i;
    for(i = 0; i < 5; i++)
    {
        array[i] = array[i] + 1; //This modifies the underlying memory
    }
    return 0; //Return 0 for OK, 1 for problem.
}

在 Python 方面，您可以通过检查 arr 来查看结果。

from ctypes import *
libCalc = CDLL("libcalci.so")
pyarr = [65, 66, 67, 68]                   #Create List with underlying memory
arr = (ctypes.c_int * len(pyarr))(*pyarr)  #Create ctypes pointer to underlying memory
res = libCalc.addOne(arr)                  #Hands over pointer to underlying memory

if res==0:
    print(', '.join(arr))                  #Output array