c++ 读取 二维 numpy

要在 C++ 中读取 NumPy 生成的二维数组，可以使用 NumPy 提供的 C API。以下是一个简单的示例代码：

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

int main()
{
    Py_Initialize();
    import_array();

    // 读取 NumPy 数组
    PyObject* array = PyArray_FromFile(stdin, NULL, NPY_BIG_ENDIAN, 2, NULL);
    if (array == NULL)
    {
        PyErr_Print();
        return 1;
    }

    // 转换为 C 数组
    PyArrayObject* np_array = reinterpret_cast<PyArrayObject*>(array);
    int rows = PyArray_DIM(np_array, 0);
    int cols = PyArray_DIM(np_array, 1);
    double* data = static_cast<double*>(PyArray_DATA(np_array));

    // 处理数据
    for (int i = 0; i < rows; ++i)
    {
        for (int j = 0; j < cols; ++j)
        {
            double value = data[i * cols + j];
            // 处理每一个元素的值
        }
    }

    // 释放资源
    Py_DECREF(np_array);
    Py_DECREF(array);
    Py_Finalize();
    return 0;
}

这个示例代码假设输入的二维 NumPy 数组是双精度浮点数类型。如果输入的类型不同，需要根据实际情况修改代码。另外，需要注意的是，这个示例代码没有进行错误处理，实际使用时需要添加相应的错误处理逻辑。