Python C 绑定 - 从 python 到 C++ 获取数组

Question

正如标题所说：我想在 C++ 中创建一个 python 绑定，对某个数组进行一些代数运算。 为此，我必须将 python“数组对象”解析为 C++ 作为双精度或整数向量或任何情况。 我试图这样做，但我面临一些问题。 我创建了一个新的 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

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

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

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;