Question

我正在尝试在我的C ++应用程序中使用python 3（带有numpy）。这需要将一个C ++数组发送到python，执行计算，然后在C ++中检索结果。为此，我基于这里讨论的代码： https://codereview.stackexchange.com/questions/92266/sending-a-c-array-to-python-numpy-and-back/92353#92353 还在这里： Sending a C++ array to Python and back (Extending C++ with Numpy)

虽然代码审查帖子中的示例基本上可行，但是当我修改python和C ++脚本时，我遇到了返回值的麻烦：当我试图返回在python中创建的变量时，结果是一个nan的向量而不是预期的计算。我的猜测是，对象在某种程度上超出了范围，但我无法解决这个问题。

我在名为 mymodule.py 的文件中使用以下python脚本：

import numpy

def array_tutorial(a):
    print("array_tutorial - python")
    print(a)
    print("")
    firstRow = a[0,:]
    #beta = numpy.array([[10,20,30],[10,20,30],[10,20,30]])
    #firstRow = beta[0,:]
    return firstRow

def myfunction():
    beta = numpy.array([[1,2,3],[1,2,3],[1,2,3]])
    print("myfunction - python")
    print(beta)
    print("")
    firstRow = beta[0,:]
    return firstRow

我的C ++代码位于 numpy_cpp.cpp 文件中，这是代码审核帖子的已接受答案的略微更改和简化版本。

#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION

#include <stdio.h>
#include <iostream>
#include <stdlib.h> 

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

int main(int argc, char* argv[])
{
    setenv("PYTHONPATH", ".", 0);

    Py_Initialize();
    import_array();

    // Build the 2D array in C++
    const int SIZE = 3;
    npy_intp dims[2]{SIZE, SIZE};
    const int ND = 2;
    long double(*c_arr)[SIZE]{ new long double[SIZE][SIZE] };

    for (int i = 0; i < SIZE; i++){
        for (int j = 0; j < SIZE; j++){
            c_arr[i][j] = i + j;}
    }

    // Convert it to a NumPy array.
    PyObject *pArray = PyArray_SimpleNewFromData(ND, dims, NPY_LONGDOUBLE, reinterpret_cast<void*>(c_arr));

    // import mymodule
    const char *module_name = "mymodule";
    PyObject *pName = PyUnicode_FromString(module_name);
    PyObject *pModule = PyImport_Import(pName);
    Py_DECREF(pName);

    // import function
    const char *func_name = "array_tutorial";
    PyObject *pFunc = PyObject_GetAttrString(pModule, func_name);
    PyObject *pReturn = PyObject_CallFunctionObjArgs(pFunc, pArray, NULL);
    PyArrayObject *np_ret = reinterpret_cast<PyArrayObject*>(pReturn);

    // Convert back to C++ array and print.
    int len = PyArray_SHAPE(np_ret)[0];
    long double* c_out;
    c_out = reinterpret_cast<long double*>(PyArray_DATA(np_ret));
    std::cout << "Printing output array - C++" << std::endl;
    for (int i = 0; i < len; i++){
        std::cout << c_out[i] << ' ';
    }
    std::cout << std::endl << std::endl;


    // import function without arguments
    const char *func_name2 = "myfunction";
    PyObject *pFunc2 = PyObject_GetAttrString(pModule, func_name2);
    PyObject *pReturn2 = PyObject_CallFunctionObjArgs(pFunc2, NULL);
    PyArrayObject *np_ret2 = reinterpret_cast<PyArrayObject*>(pReturn2);

    // convert back to C++ array and print
    int len2 = PyArray_SHAPE(np_ret2)[0];
    long double* c_out2;
    c_out2 = reinterpret_cast<long double*>(PyArray_DATA(np_ret2));
    std::cout << "Printing output array 2 - C++" << std::endl;
    for (int i = 0; i < len2; i++){
        std::cout << c_out2[i] << ' ';
    }
    std::cout << std::endl << std::endl;

    Py_Finalize();
    return 0;
}

与接受的答案相比，我必须添加

setenv("PYTHONPATH", ".", 0);

为了确保找到python脚本，我为函数＆＃34; myfunction＆＃34;添加了第二个函数调用。没有输入参数，我删除了一些错误处理。

我在Ubuntu 16.10上并使用

g++ -Wall numpy_cpp.cpp -I/usr/include/python3.5m/ -lpython3.5m

编译和链接（除了import_array（）的一个警告之外没什么问题）。我的目标是python 3。

然后运行程序会提供以下控制台输出：

array_tutorial - python
[[ 0.0  1.0  2.0]
 [ 1.0  2.0  3.0]
 [ 2.0  3.0  4.0]]

Printing output array - C++
0 1 2 

myfunction - python
[[1 2 3]
 [1 2 3]
 [1 2 3]]

Printing output array 2 - C++
nan nan nan

这是给我带来麻烦的最后一个输出，其中python返回在python脚本中设置的numpy数组的第一行。从python print语句来看，似乎numpy数组很好（不是nan），但是一旦它被引用到C ++，事情就会横向发展。

如果我取消注释array_tutorial函数中return语句上面的两行，我得到第一个函数调用的相同（令人失望的）结果。

因此，我的问题是如何在没有对象（可能）超出范围的情况下在C ++中获取正确的值？

我为冗长的帖子道歉，并提前感谢您的帮助！

编辑：正如下面的lomereiter所指出的，应该设置python中的numpy数组，并记住数据类型。这解决了这个问题。一个更好的python脚本输出接收数组的数据类型并指定声明的数组的数据类型：

import numpy

def array_tutorial(a):
    print("array_tutorial - python")
    print(a)
    print(numpy.dtype(a[0,0]))
    print("")
    firstRow = a[0,:]
    #beta = numpy.array([[10,20,30],[10,20,30],[10,20,30]],dtype=numpy.float128)
    #firstRow = beta[0,:]
    return firstRow

def myfunction():
    beta = numpy.array([[1,2,3],[1,2,3],[1,2,3]],dtype=numpy.float128)
    print("myfunction - python")
    print(beta)
    print("")
    firstRow = beta[0,:]
    return firstRow

Answer 1

在Python代码中创建数组时，应指定dtype。您正在使用C ++代码加长两倍，而dtype被推断为int64（在64位平台上）

在C ++中嵌入python / numpy

1 个答案: