如何在Python C扩展中将动态分配的C数组转为Numpy数组并返回

核心问题：数组索引计算错误 #

循环填充内存数组时，索引公式错误，导致数据在内存中的排列和Numpy预期的维度布局不匹配。虽然打印时能读取到正确填充的位置，但Numpy按声明的(2, rows, columns)维度访问内存时，读取到的是未正确填充的区域，因此出现乱值。

具体错误点及修复 #

1. 循环索引公式错误 #

原代码中使用rr * rows + cc计算索引，混淆了行数和列数的作用。对于形状为(rows, columns)的输入，按C顺序（行优先）排列的2D数组，正确的索引应为rr * columns + cc（每行包含columns个元素）。

修复后的循环填充代码：

for (size_t cc = 0; cc < columns; ++cc) {
    // some calculation
    for (size_t rr = 0; rr < rows; ++rr) {
        // some calculation
        size_t idx = rr * columns + cc; // 修正索引计算
        r_arr[idx] = result1;
        r_arr[idx + rows * columns] = result2; // 第二组数据偏移正确
        PySys_WriteStdout("%f, %f\n", r_arr[idx], r_arr[idx + rows * columns]);
    }
}

2. 内存分配失败后的错误处理不完整 #

原代码在malloc失败后仅释放了输入数组的引用，但未终止函数执行，会导致后续代码继续访问空指针，引发未定义行为。

修复后的内存分配代码：

double* r_arr = (double*)malloc((rows * columns * 2) * sizeof(double));
if (r_arr == NULL) {
    PyErr_SetString(PyExc_ValueError, "Failed to allocate memory to arrays.");
    Py_DECREF(data_array);
    return NULL; // 必须返回NULL终止执行
}

3. PyErr_Print调用错误 #

PyErr_Print不需要传入参数，原代码的调用方式不符合API规范。

修复后的错误处理代码：

if (r_obj == NULL) {
    PyErr_Print(); // 正确调用方式
    return NULL;
}

验证说明 #

修复后，内存数组的排列顺序将完全匹配Numpy数组(2, rows, columns)的C顺序布局：

• 第一部分（索引0到rows*columns-1）对应输出数组的第一个维度（[0, :, :]）
• 第二部分（索引rows*columns到2*rows*columns-1）对应输出数组的第二个维度（[1, :, :]）

此时返回的Numpy数组将正确读取填充的数据，不再出现乱值。

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