在Visual Studio VC++中使用Numpy arrayobject：数组转换及矩阵乘法问题

原代码的核心问题 #

• 错误调用PyArray_Type执行运算：PyArray_Type是Numpy数组的类型元对象，并非乘法函数
• 逻辑偏离需求：原代码尝试实现数组标量乘法，但目标是两个Numpy数组的矩阵乘法
• 缺失错误检查：无法定位初始化、对象创建、函数调用阶段的问题
• 仅支持一维数组：未满足矩阵乘法的二维数组维度要求

修复后的完整代码 #

#include <Python.h>
#include <numpy/arrayobject.h>
#include <iostream>

int main(int argc, char* argv[])
{
    // 初始化Python解释器
    if (!Py_Initialize()) {
        std::cerr << "Python初始化失败" << std::endl;
        return -1;
    }

    // 初始化Numpy C API
    import_array();
    if (_import_array() < 0) {
        std::cerr << "Numpy API初始化失败" << std::endl;
        Py_Finalize();
        return -1;
    }

    // 创建第一个矩阵 (2x3)
    int data1[6] = {1,2,3,4,5,6};
    npy_intp dims1[2] = {2, 3};
    PyArrayObject* arr1 = (PyArrayObject*)PyArray_SimpleNewFromData(2, dims1, NPY_INT, data1);
    if (!arr1) {
        std::cerr << "创建第一个Numpy数组失败" << std::endl;
        Py_Finalize();
        return -1;
    }

    // 创建第二个矩阵 (3x2)
    int data2[6] = {7,8,9,10,11,12};
    npy_intp dims2[2] = {3, 2};
    PyArrayObject* arr2 = (PyArrayObject*)PyArray_SimpleNewFromData(2, dims2, NPY_INT, data2);
    if (!arr2) {
        std::cerr << "创建第二个Numpy数组失败" << std::endl;
        Py_DECREF(arr1);
        Py_Finalize();
        return -1;
    }

    // 获取Numpy的矩阵乘法函数(dot)
    PyObject* numpy_module = PyImport_ImportModule("numpy");
    if (!numpy_module) {
        std::cerr << "导入numpy模块失败" << std::endl;
        Py_DECREF(arr1);
        Py_DECREF(arr2);
        Py_Finalize();
        return -1;
    }

    PyObject* dot_func = PyObject_GetAttrString(numpy_module, "dot");
    if (!dot_func || !PyCallable_Check(dot_func)) {
        std::cerr << "获取numpy.dot函数失败" << std::endl;
        Py_DECREF(numpy_module);
        Py_DECREF(arr1);
        Py_DECREF(arr2);
        Py_Finalize();
        return -1;
    }

    // 调用矩阵乘法函数
    PyObject* args = PyTuple_Pack(2, (PyObject*)arr1, (PyObject*)arr2);
    PyObject* result = PyObject_CallObject(dot_func, args);
    if (!result) {
        std::cerr << "矩阵乘法调用失败" << std::endl;
        Py_DECREF(args);
        Py_DECREF(dot_func);
        Py_DECREF(numpy_module);
        Py_DECREF(arr1);
        Py_DECREF(arr2);
        Py_Finalize();
        return -1;
    }

    // 输出结果矩阵 (2x2)
    if (PyArray_Check(result)) {
        PyArrayObject* result_arr = (PyArrayObject*)result;
        int* result_data = (int*)PyArray_DATA(result_arr);
        npy_intp rows = PyArray_DIM(result_arr, 0);
        npy_intp cols = PyArray_DIM(result_arr, 1);

        std::cout << "矩阵乘法结果:" << std::endl;
        for (npy_intp i = 0; i < rows; ++i) {
            for (npy_intp j = 0; j < cols; ++j) {
                std::cout << result_data[i * cols + j] << "\t";
            }
            std::cout << std::endl;
        }
    }

    // 释放所有Python对象引用
    Py_DECREF(result);
    Py_DECREF(args);
    Py_DECREF(dot_func);
    Py_DECREF(numpy_module);
    Py_DECREF(arr1);
    Py_DECREF(arr2);

    Py_Finalize();
    return 0;
}

VC++项目配置要点 #

• 包含目录：添加Python的include目录和Numpy的core/include目录（示例路径：C:\PythonXX\include、C:\PythonXX\Lib\site-packages\numpy\core\include）
• 库目录：添加Python的libs目录（示例路径：C:\PythonXX\libs）
• 附加依赖项：链接对应版本的Python库（示例：python310.lib，XX为Python版本号）
• 位数匹配：确保项目编译位数（x86/x64）与Python安装位数一致，否则会出现链接或运行错误

关键修正说明 #

1. 正确获取运算函数：通过PyImport_ImportModule导入numpy模块，再获取dot函数实现矩阵乘法
2. 支持二维数组：创建符合矩阵乘法维度要求的二维数组（第一个矩阵列数=第二个矩阵行数）
3. 完善错误检查：对每一步Python API调用的返回值做有效性判断，便于排查问题
4. 规范内存管理：所有创建的PyObject对象都调用Py_DECREF释放，避免内存泄漏

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