C调用Python模块后，如何解析传回的列表并存储到C数组/vector

我之前刚好处理过类似的C调用Python模块解析列表的场景，给你分享下具体的实现思路和方案吧！

首先明确：完全不需要把列表从元组中分离单独返回，也不用拆分成多个函数——直接在C代码里通过Python C API解析元组内的列表即可，逻辑清晰且能避免多次跨语言调用的开销。下面分几个部分详细说明：

一、核心实现：解析元组中的Python列表到C数组/vector #

Python C API提供了完整的列表操作函数，按以下步骤就能完成解析：

1. 先解析元组拿到列表对象 #

用PyArg_ParseTuple时，用O格式符获取列表对应的PyObject*对象（O表示接收任意Python对象）。比如假设Python返回的元组结构是(整数, 整数列表, 字符串)，解析代码如下：

PyObject* list_obj;
int num;
const char* str;
if (!PyArg_ParseTuple(pResult, "iOs", &num, &list_obj, &str)) {
    PyErr_Print();
    // 记得清理已获取的Python资源
    return 1;
}

2. 验证列表类型并获取长度 #

拿到对象后，先检查是否是合法的列表，再获取其长度：

if (!PyList_Check(list_obj)) {
    PyErr_SetString(PyExc_TypeError, "Tuple must contain a list");
    PyErr_Print();
    // 资源清理...
    return 1;
}
Py_ssize_t list_len = PyList_Size(list_obj);

3. 遍历列表转换为C类型，存入数组/vector #

循环取出每个列表元素，转换为对应的C类型（比如整数转long，字符串转char*），这里以存入std::vector<long>为例：

#include <vector>

std::vector<long> c_vec;
c_vec.reserve(list_len); // 预分配内存提升效率

for (Py_ssize_t i = 0; i < list_len; ++i) {
    // PyList_GetItem返回的是"借用引用"，不需要手动调用Py_DECREF
    PyObject* item = PyList_GetItem(list_obj, i);
    
    // 先检查元素类型是否符合预期（这里假设是整数）
    if (!PyLong_Check(item)) {
        PyErr_SetString(PyExc_TypeError, "List items must be integers");
        PyErr_Print();
        // 资源清理...
        return 1;
    }
    
    // 转换为C类型，同时处理转换错误
    long val = PyLong_AsLong(item);
    if (val == -1 && PyErr_Occurred()) {
        PyErr_Print();
        // 资源清理...
        return 1;
    }
    
    c_vec.push_back(val);
}

如果要存入C数组，需要先动态分配内存（比如long* c_arr = (long*)malloc(list_len * sizeof(long));），循环赋值后记得在不用时free(c_arr)。

二、关于返回方式的选择：不用拆分函数 #

你提到的“拆分多个函数返回”完全没必要，理由如下：

• 跨语言调用开销高：C和Python之间的函数调用涉及上下文切换，一次返回所有关联数据比多次调用高效得多。
• 逻辑更清晰：如果列表和元组里的其他数据是强关联的（比如一组统计结果包含总数、明细列表、备注），放在一个元组返回更符合语义。
• 解析成本低：解析列表的代码量很小，完全不需要为了“方便解析”而拆分返回。

只有当以下场景时，才考虑拆分：

• 列表数据量极大，单独返回可以分批处理；
• 列表和其他数据的更新逻辑完全独立，后续可能需要单独获取某部分数据。

三、关键注意事项 #

1. 内存管理：PyList_GetItem返回的是借用的引用，不需要调用Py_DECREF；如果是自己创建的Python对象（比如用PyLong_FromLong创建的）才需要手动释放。
2. 类型严格检查：一定要验证列表元素的类型，避免转换时出现未定义行为。
3. 错误处理：每一步转换都要检查是否触发了Python异常（比如PyErr_Occurred()），出现异常要及时清理资源并返回。

完整示例代码 #

Python模块（mymod.py） #

def get_combined_data():
    # 返回元组：(状态码, 数值列表, 描述信息)
    return (0, [15, 25, 35, 45, 55], "success")

C调用代码 #

#include <Python.h>
#include <vector>
#include <stdio.h>

int main() {
    // 初始化Python解释器
    if (!Py_Initialize()) {
        fprintf(stderr, "Failed to initialize Python\n");
        return 1;
    }

    // 加载模块
    PyObject* pModule = PyImport_ImportModule("mymod");
    if (!pModule) {
        PyErr_Print();
        Py_Finalize();
        return 1;
    }

    // 获取目标函数
    PyObject* pFunc = PyObject_GetAttrString(pModule, "get_combined_data");
    if (!pFunc || !PyCallable_Check(pFunc)) {
        PyErr_Print();
        Py_DECREF(pModule);
        Py_Finalize();
        return 1;
    }

    // 调用函数（无参数）
    PyObject* pResult = PyObject_CallObject(pFunc, NULL);
    Py_DECREF(pFunc);
    if (!pResult) {
        PyErr_Print();
        Py_DECREF(pModule);
        Py_Finalize();
        return 1;
    }

    // 解析元组
    PyObject* list_obj;
    int status;
    const char* msg;
    if (!PyArg_ParseTuple(pResult, "iOs", &status, &list_obj, &msg)) {
        PyErr_Print();
        Py_DECREF(pResult);
        Py_DECREF(pModule);
        Py_Finalize();
        return 1;
    }

    // 解析列表到vector
    std::vector<long> c_data_list;
    if (PyList_Check(list_obj)) {
        Py_ssize_t len = PyList_Size(list_obj);
        c_data_list.reserve(len);
        for (Py_ssize_t i = 0; i < len; ++i) {
            PyObject* item = PyList_GetItem(list_obj, i);
            if (PyLong_Check(item)) {
                long val = PyLong_AsLong(item);
                if (val == -1 && PyErr_Occurred()) {
                    PyErr_Print();
                    goto cleanup;
                }
                c_data_list.push_back(val);
            } else {
                PyErr_SetString(PyExc_TypeError, "List items must be integers");
                PyErr_Print();
                goto cleanup;
            }
        }
    } else {
        PyErr_SetString(PyExc_TypeError, "Expected a list in result tuple");
        PyErr_Print();
        goto cleanup;
    }

    // 打印结果
    printf("Status: %d\n", status);
    printf("Message: %s\n", msg);
    printf("List data (length %zu): ", c_data_list.size());
    for (long val : c_data_list) {
        printf("%ld ", val);
    }
    printf("\n");

cleanup:
    // 清理资源
    Py_DECREF(pResult);
    Py_DECREF(pModule);
    Py_Finalize();
    return 0;
}

内容的提问来源于stack exchange，提问作者nice_remark