Python C 绑定(bind) - 将数组从 python 获取到 C++

Question

正如标题所说:我想在 C++ 中创建一个 python 绑定(bind)，对某个数组执行一些代数运算。为此，我必须将 python“数组对象”解析为 C++ 作为 double 或整数或任何情况的 vector 。我尝试这样做，但遇到了一些问题。我创建了一个新的 python 类型和一个名为 Typer 的类，其中我有一个方法尝试获取 python 数组的元素，然后计算总和(作为第一步)。

tatic PyObject *Typer_vectorsum(Typer *self, PyObject *args)
{
    PyObject *retval;
    PyObject *list;

    if (!PyArg_ParseTuple(args, "O", &list))
        return NULL;
    double *arr;
    arr = (double *)malloc(sizeof(double) * PyTuple_Size(list));
    int length;
    length = PyTuple_Size(list);
    PyObject *item = NULL;
    for (int i = 0; i < length; ++i)
    {
        item = PyTuple_GetItem(list, i);
        if (!PyFloat_Check(item))
        {
            exit(1);
        }
        arr[i] = PyFloat_AsDouble(item);
    }
    double result = 0.0;
    for (int i = 0; i < length; ++i)
    {
        result += arr[i];
    }
    retval = PyFloat_FromDouble(result);
    free(arr);
    return retval;
}

在这个方法中，我将Python数组对象解析为C数组(使用malloc分配数组的内存)。然后，我将对象中的每个元素添加到我的 C 数组中，并在最后一个 for 循环中计算总和。

如果我构建项目，然后创建一个 python 测试文件，则不会发生任何情况(该文件编译时没有任何问题，但不打印任何内容)。

y = example.Typer() .  #typer is the init
tuple = (1, 2, 3)
print(y.vectorsum(tuple))

我错过了什么吗？另外，是否有一种简单易用的方法将 python 数组对象转换为 C++，但作为 std::vector 而不是经典的 C 数组？

提前谢谢您!

Answer 1

元组包含整数，而不是 float ，因此您的 PyFloat_Check 失败。不，没有从 Python 元组到 C 数组或 C++ std::vector 的直接方法。原因是元组是 Python 对象的数组，而不是 C 值的数组，例如 doubles。

这是您的示例，改进了错误检查，之后它应该可以工作:

PyObject *retval;
PyObject *list;

if (!PyArg_ParseTuple(args, "O!", &PyTuple_Type, &list))
    return NULL;

double *arr = 
arr = (double *)malloc(sizeof(double) * PyTuple_GET_SIZE(list));
int length;
length = PyTuple_GET_SIZE(list);
PyObject *item = NULL;
for (int i = 0; i < length; ++i)
{   
    item = PyTuple_GET_ITEM(list, i);
    arr[i] = PyFloat_AsDouble(item);
    if (arr[i] == -1. && PyErr_Occurred())
    {   
        exit(1);
    }
}
double result = 0.0;
for (int i = 0; i < length; ++i)
{
    result += arr[i];
}
retval = PyFloat_FromDouble(result);
free(arr);
return retval;