返回Cython数组

Question

如何正确初始化并返回Cython数组？例如：

cdef public double* cyTest(double[] input):
  cdef double output[3]

  for i in xrange(3):
    output[i] = input[i]**2
    print 'loop: ' + str(output[i])
  return output

cdef double* test = [1,2,3]
cdef double* results = cyTest(test)

for i in xrange(3):
  print 'return: ' + str(results[i])

这将返回：

^{pr2}$

显然，results仍然只指向垃圾，而不是它应该指向的值。诚然，我对混合指针和数组语法以及在Cython上下文中哪一种更可取/可能有点困惑。在

最后，我想从纯C++函数调用{{CD2}}：

#include <iostream>
#include <Python.h>
#include "cyTest.h"

void main() {
  Py_Initialize();
  initcyTest();
  double input[3] = {1,2,3};
  double* output = cyTest(input);

  for(int i = 0; i < 3; i++)
    std::cout << "cout: " << output[i] << std::endl;

  Py_Finalize();
}

这将返回类似的结果：

loop: 1.0->1.0
loop: 2.0->4.0
loop: 3.0->9.0
cout: 1
cout: 6.30058e+077
cout: 6.39301e-308

有人能解释一下我犯了什么错误吗？我想尽量简单些。它只是将数组从Cython返回到C++。如果不需要的话，稍后我将处理动态内存分配。在

Answer 1

您正在返回对本地数组（output）的引用，这将不起作用。在

尝试将脚本更改为：

from cpython.mem cimport PyMem_Malloc

cdef public double * cyTest(double[] input):
    cdef double * output = < double * >PyMem_Malloc( sizeof(double) * 3 )
    for i in xrange(3):
        output[i] = input[i]**2
        print 'loop: ' + str(output[i])
    return output

在你的c++代码中

使用完double* output问题free( output );

如果您想在pyx脚本中使用cdef double* results = cyTest(test)，那么不要忘记使用PyMem_Free(results)